xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-08 20:34:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model_loading.py

Browse Source
This commit is contained in:
kijai 2024-11-20 21:28:53 +02:00
parent e5fc7c1bf3
commit e52dc36bc5
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
model_loading.py
View File

@ -270,13 +270,13 @@ class DownloadAndLoadCogVideoModel:
transformer = merge_lora(transformer, l["path"], l["strength"]) transformer = merge_lora(transformer, l["path"], l["strength"])
except: except:
raise ValueError(f"Can't recognize LoRA {l['path']}") raise ValueError(f"Can't recognize LoRA {l['path']}")
if adapter_list:
pipe.set_adapters(adapter_list, adapter_weights=adapter_weights) pipe.set_adapters(adapter_list, adapter_weights=adapter_weights)
if fuse: if fuse:
lora_scale = 1 lora_scale = 1
if dimensionx_lora: if dimensionx_lora:
lora_scale = lora_scale / lora_rank lora_scale = lora_scale / lora_rank
pipe.fuse_lora(lora_scale=lora_scale, components=["transformer"]) pipe.fuse_lora(lora_scale=lora_scale, components=["transformer"])
if "fused" in attention_mode: if "fused" in attention_mode: