updates

huggingface · Dec 23, 2024 · 4c5082c · 4c5082c
1 parent fb17ba3
commit 4c5082c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/diffusers/pipelines/flux/pipeline_flux_control.py b/src/diffusers/pipelines/flux/pipeline_flux_control.py
@@ -575,6 +575,7 @@ def prepare_image(
         if isinstance(image, torch.Tensor):
             pass
         else:
+            print(f"{type(image)=}")
             image = self.image_processor.preprocess(image, height=height, width=width)
 
         image_batch_size = image.shape[0]

diff --git a/tests/lora/utils.py b/tests/lora/utils.py
@@ -1568,7 +1568,7 @@ def test_lora_fuse_nan(self):
 
             # without we should not see an error, but every image will be black
             pipe.fuse_lora(components=self.pipeline_class._lora_loadable_modules, safe_fusing=False)
-            out = pipe("test", num_inference_steps=2, output_type="np")[0]
+            out = pipe(**inputs)[0]
 
             self.assertTrue(np.isnan(out).all())