1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Fixed minor error in test_lora_layers_peft.py (#7394)

* Update test_lora_layers_peft.py

* Update utils.py
This commit is contained in:
UmerHA
2024-03-25 22:35:27 +01:00
committed by GitHub
parent a51b6cc86a
commit 1cd4732e7f

View File

@@ -73,7 +73,7 @@ class PeftLoraLoaderMixinTests:
vae_kwargs = 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
torch.manual_seed(0)