fixes

huggingface · Jan 13, 2025 · 96eced3 · 96eced3
1 parent b2afc10
commit 96eced3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/lora/utils.py b/tests/lora/utils.py
@@ -1932,11 +1932,11 @@ def test_logs_info_when_no_lora_keys_found(self):
             if "text_encoder" in lora_module:
                 text_encoder = getattr(pipe, lora_module)
                 if lora_module == "text_encoder":
-                    prefix = text_encoder
+                    prefix = "text_encoder"
                 elif lora_module == "text_encoder_2":
                     prefix = "text_encoder_2"
 
-                logger = logging.get_logger("diffusers.loaders.lora_pipeline")
+                logger = logging.get_logger("diffusers.loaders.lora_base")
                 logger.setLevel(logging.INFO)
 
                 with CaptureLogger(logger) as cap_logger: