diff --git a/tests/lora/utils.py b/tests/lora/utils.py index 567b79677ffd..07563a84b5a6 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())