use with load lora weight

zRzRzRzRzRzRzR · zRzRzRzRzRzRzR · commit 6fde85f0da04 · 2025-03-30T12:59:08.000+08:00
diff --git a/src/cogkit/utils/lora.py b/src/cogkit/utils/lora.py
@@ -9,8 +9,8 @@ def load_lora_checkpoint(
     lora_model_id_or_path: str,
     lora_scale: float = 1.0,
 ) -> None:
-    pipeline.load_lora_weights(lora_model_id_or_path)
-    pipeline.fuse_lora(components=["transformer"], lora_scale=lora_scale)
+    pipeline.load_lora_weights(lora_model_id_or_path, lora_scale=lora_scale)
+    # pipeline.fuse_lora(components=["transformer"], lora_scale=lora_scale)
 
 
 def unload_lora_checkpoint(
