From 1cd4732e7fd2b5b9e31d124fd437dcb3d168a3ba Mon Sep 17 00:00:00 2001
From: UmerHA <40663591+UmerHA@users.noreply.github.com>
Date: Mon, 25 Mar 2024 22:35:27 +0100
Subject: [PATCH] Fixed minor error in `test_lora_layers_peft.py` (#7394)

* Update test_lora_layers_peft.py

* Update utils.py
---
 tests/lora/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)