Skip to content

GitLab

Unverified Commit ea1ba0ba authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[LoRA] test fix (#10351) 

updates
parent 9d27df80
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/lora/utils.py
View file @ ea1ba0ba
...@@ -1568,7 +1568,7 @@ class PeftLoraLoaderMixinTests: ...@@ -1568,7 +1568,7 @@ class PeftLoraLoaderMixinTests:
# without we should not see an error, but every image will be black # 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) 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()) self.assertTrue(np.isnan(out).all())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment