From ea1ba0ba53bdd6569547e26e518f094745ed9d03 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Mon, 23 Dec 2024 15:45:45 +0530 Subject: [PATCH] [LoRA] test fix (#10351) 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 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())