feat: P mode in image

piercus · Jan 15, 2024 · a05aaee · a05aaee
1 parent 0c73c93
commit a05aaee
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/refiners/fluxion/utils.py b/src/refiners/fluxion/utils.py
@@ -121,6 +121,9 @@ def image_to_tensor(image: Image.Image, device: Device | str | None = None, dtyp
 
     Values are clamped to the range `[0, 1]`.
     """
+    if image.mode == 'P':
+        image = image.convert('RGB')
+
     image_tensor = torch.tensor(array(image).astype(float32) / 255.0, device=device, dtype=dtype)
 
     match image.mode: