From 6038b2183014913e50694d80f1a78414c8b82e73 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 15 Sep 2024 17:13:01 +0300 Subject: [PATCH] Add bertjiazheng/KoolCogVideoX-5b -finetune https://huggingface.co/bertjiazheng/KoolCogVideoX-5b --- nodes.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 9053168..0c6ff19 100644 --- a/nodes.py +++ b/nodes.py @@ -23,6 +23,7 @@ class DownloadAndLoadCogVideoModel: [ "THUDM/CogVideoX-2b", "THUDM/CogVideoX-5b", + "bertjiazheng/KoolCogVideoX-5b", ], ), @@ -52,7 +53,7 @@ class DownloadAndLoadCogVideoModel: if "2b" in model: base_path = os.path.join(folder_paths.models_dir, "CogVideo", "CogVideo2B") 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): log.info(f"Downloading model to: {base_path}")