mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix lora loading with diffusers 0.33.1

Browse Source
This commit is contained in:
rkfg
2025-05-19 09:33:56 +03:00
parent 74121a5d8a
commit 481d950db4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/pipeline_ace_step.py
+1 -1
View File
@@ -1634,7 +1634,7 @@ class ACEStepPipeline:
lora_download_path = lora_name_or_path lora_download_path = lora_name_or_path
if self.lora_path != "none": if self.lora_path != "none":
self.ace_step_transformer.unload_lora() self.ace_step_transformer.unload_lora()
self.ace_step_transformer.load_lora_adapter(os.path.join(lora_download_path, "pytorch_lora_weights.safetensors"), adapter_name="zh_rap_lora", with_alpha=True) self.ace_step_transformer.load_lora_adapter(os.path.join(lora_download_path, "pytorch_lora_weights.safetensors"), adapter_name="zh_rap_lora", with_alpha=True, prefix=None)
logger.info(f"Loading lora weights from: {lora_name_or_path} download path is: {lora_download_path}") logger.info(f"Loading lora weights from: {lora_name_or_path} download path is: {lora_download_path}")
self.lora_path = lora_name_or_path self.lora_path = lora_name_or_path
elif self.lora_path != "none" and lora_name_or_path == "none": elif self.lora_path != "none" and lora_name_or_path == "none":