From d9b9ffab68d792eeef5cbf6c958bb98116fd6163 Mon Sep 17 00:00:00 2001 From: DvirYBria <74497787+DvirYBria@users.noreply.github.com> Date: Sun, 1 Dec 2024 11:21:46 +0200 Subject: [PATCH] Update bria_api_node.py RGBA -> RGB --- bria_api_node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bria_api_node.py b/bria_api_node.py index 00f6c04..2e4c49a 100644 --- a/bria_api_node.py +++ b/bria_api_node.py @@ -174,7 +174,7 @@ class GenerativeFillNode(BriaAPINode): response_dict = response.json() image_response = requests.get(response_dict['urls'][0]) result_image = Image.open(io.BytesIO(image_response.content)) - result_image = result_image.convert("RGBA") + result_image = result_image.convert("RGB") result_image = np.array(result_image).astype(np.float32) / 255.0 result_image = torch.from_numpy(result_image)[None,] return (result_image,)