diff --git a/tests/lora/utils.py b/tests/lora/utils.py index da0da0ad8a..6be602a77b 100644 --- a/tests/lora/utils.py +++ b/tests/lora/utils.py @@ -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)