Skip to content

Commit 2a3547e

Browse files
committed
fix
1 parent 4e2cc1c commit 2a3547e

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/lora/utils.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -877,15 +877,11 @@ def test_simple_inference_with_text_denoiser_lora_unfused(
877877
pipe, denoiser = self.check_if_adapters_added_correctly(pipe, text_lora_config, denoiser_lora_config)
878878

879879
pipe.fuse_lora(components=self.pipeline_class._lora_loadable_modules)
880-
self.assertTrue(
881-
pipe.num_fused_loras == 1, pipe.num_fused_loras, f"{pipe.num_fused_loras=}, {pipe.fused_loras=}"
882-
)
880+
self.assertTrue(pipe.num_fused_loras == 1, f"{pipe.num_fused_loras=}, {pipe.fused_loras=}")
883881
output_fused_lora = pipe(**inputs, generator=torch.manual_seed(0))[0]
884882

885883
pipe.unfuse_lora(components=self.pipeline_class._lora_loadable_modules)
886-
self.assertTrue(
887-
pipe.num_fused_loras == 0, pipe.num_fused_loras, f"{pipe.num_fused_loras=}, {pipe.fused_loras=}"
888-
)
884+
self.assertTrue(pipe.num_fused_loras == 0, f"{pipe.num_fused_loras=}, {pipe.fused_loras=}")
889885
output_unfused_lora = pipe(**inputs, generator=torch.manual_seed(0))[0]
890886

891887
# unloading should remove the LoRA layers

0 commit comments

Comments
 (0)