diff --git a/tests/lora/utils.py b/tests/lora/utils.py index 0a0366fd8d2b..6cfa0416d4e2 100644 --- 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())