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

Add bertjiazheng/KoolCogVideoX-5b -finetune

Browse Source 
https://huggingface.co/bertjiazheng/KoolCogVideoX-5b
This commit is contained in:
kijai 2024-09-15 17:13:01 +03:00
parent 05413b6cfe
commit 6038b21830
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes.py
View File

@ -23,6 +23,7 @@ class DownloadAndLoadCogVideoModel:
[ [
"THUDM/CogVideoX-2b", "THUDM/CogVideoX-2b",
"THUDM/CogVideoX-5b", "THUDM/CogVideoX-5b",
"bertjiazheng/KoolCogVideoX-5b",
], ],
), ),
@ -52,7 +53,7 @@ class DownloadAndLoadCogVideoModel:
if "2b" in model: if "2b" in model:
base_path = os.path.join(folder_paths.models_dir, "CogVideo", "CogVideo2B") base_path = os.path.join(folder_paths.models_dir, "CogVideo", "CogVideo2B")
elif "5b" in model: elif "5b" in model:
base_path = os.path.join(folder_paths.models_dir, "CogVideo", "CogVideoX-5b") base_path = os.path.join(folder_paths.models_dir, "CogVideo", (model.split("/")[-1]))
if not os.path.exists(base_path): if not os.path.exists(base_path):
log.info(f"Downloading model to: {base_path}") log.info(f"Downloading model to: {base_path}")