From 49abc7b0ebf423d732a89182bbc0c111170e15df Mon Sep 17 00:00:00 2001 From: caojiachen1 Date: Sat, 7 Sep 2024 13:14:18 +0800 Subject: [PATCH] Rectify model path --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 9053168..5add6e4 100644 --- a/nodes.py +++ b/nodes.py @@ -50,7 +50,7 @@ class DownloadAndLoadCogVideoModel: dtype = {"bf16": torch.bfloat16, "fp16": torch.float16, "fp32": torch.float32}[precision] 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", "CogVideoX-2b") elif "5b" in model: base_path = os.path.join(folder_paths.models_dir, "CogVideo", "CogVideoX-5b")