Skip to content

GitLab

Unverified Commit 1cd4732e authored by UmerHA's avatar UmerHA Committed by GitHub
Browse files

Fixed minor error in `test_lora_layers_peft.py` (#7394) 

* Update test_lora_layers_peft.py

* Update utils.py
parent a51b6cc8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/lora/utils.py
View file @ 1cd4732e
...@@ -73,7 +73,7 @@ class PeftLoraLoaderMixinTests: ...@@ -73,7 +73,7 @@ class PeftLoraLoaderMixinTests:
vae_kwargs = None vae_kwargs = None
def get_dummy_components(self, scheduler_cls=None): def get_dummy_components(self, scheduler_cls=None):
scheduler_cls = self.scheduler_cls if scheduler_cls is None else LCMScheduler scheduler_cls = self.scheduler_cls if scheduler_cls is None else scheduler_cls
rank = 4 rank = 4
torch.manual_seed(0) torch.manual_seed(0)
......
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