From 5a5a80370836928a7063fe017ab645f6e51a98bb Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 23 Dec 2024 13:03:19 +0530 Subject: [PATCH] updates --- tests/lora/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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())