[LoRA] test fix (#10351)

updates

[LoRA] test fix (#10351)
updates
ea1ba0ba · Sayak Paul · GitHub · 9d27df80 · ea1ba0ba
Unverified Commit ea1ba0ba authored Dec 23, 2024 by Sayak Paul Committed by GitHub Dec 23, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/lora/utils.py tests/lora/utils.py +1 -1

No files found.
--- a/tests/lora/utils.py
+++ b/tests/lora/utils.py
@@ -1568,7 +1568,7 @@ class PeftLoraLoaderMixinTests:

            # 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())