Spaces:

Skywork
/

UniPic2-Metaquery

Running on Zero

yichenchenchen commited on 4 days ago

Commit

a1824bc

verified ·

1 Parent(s): f0c565b

Update inferencer.py

Files changed (1) hide show

inferencer.py CHANGED Viewed

@@ -300,6 +300,8 @@ class UniPicV2Inferencer:
         guidance_scale: float = 3.0,
         seed: int = 42
     ) -> Image.Image:
         if not self.pipeline:
             self.pipeline = self._init_pipeline()
         original_size = image.size

         guidance_scale: float = 3.0,
         seed: int = 42
     ) -> Image.Image:
+        if image.mode in ["RGBA", "LA"] or image.mode.startswith("A"):
+            image = image.convert("RGB")
         if not self.pipeline:
             self.pipeline = self._init_pipeline()
         original_size = image.size