Skip to content

Commit 1cd4732

Browse files
authored
Fixed minor error in test_lora_layers_peft.py (huggingface#7394)
* Update test_lora_layers_peft.py * Update utils.py
1 parent a51b6cc commit 1cd4732

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/lora/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ class PeftLoraLoaderMixinTests:
7373
vae_kwargs = None
7474

7575
def get_dummy_components(self, scheduler_cls=None):
76-
scheduler_cls = self.scheduler_cls if scheduler_cls is None else LCMScheduler
76+
scheduler_cls = self.scheduler_cls if scheduler_cls is None else scheduler_cls
7777
rank = 4
7878

7979
torch.manual_seed(0)

0 commit comments

Comments
 (0)