Skip to content

Commit 4c5082c

Browse files
committed
updates
1 parent fb17ba3 commit 4c5082c

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/diffusers/pipelines/flux/pipeline_flux_control.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -575,6 +575,7 @@ def prepare_image(
575575
if isinstance(image, torch.Tensor):
576576
pass
577577
else:
578+
print(f"{type(image)=}")
578579
image = self.image_processor.preprocess(image, height=height, width=width)
579580

580581
image_batch_size = image.shape[0]

tests/lora/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1568,7 +1568,7 @@ def test_lora_fuse_nan(self):
15681568

15691569
# without we should not see an error, but every image will be black
15701570
pipe.fuse_lora(components=self.pipeline_class._lora_loadable_modules, safe_fusing=False)
1571-
out = pipe("test", num_inference_steps=2, output_type="np")[0]
1571+
out = pipe(**inputs)[0]
15721572

15731573
self.assertTrue(np.isnan(out).all())
15741574

0 commit comments

Comments
 (0)