Skip to content

Commit ea1ba0b

Browse files
authored
[LoRA] test fix (#10351)
updates
1 parent 9d27df8 commit ea1ba0b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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)