nodes/aigc-apps-VideoX-Fun
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug in comfyui path

Browse Source
This commit is contained in:
bubbliiiing
2024-09-18 17:47:54 +08:00
parent 2041d1aafc
commit 24973afceb
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfyui/README.md
+1 -1
View File
@@ -26,7 +26,7 @@ cd CogVideoX-Fun/
python install.py
```
### 2. Download models into `ComfyUI/models/CogVideoX-Fun/`
### 2. Download models into `ComfyUI/models/CogVideoX_Fun/`
| Name | Storage Space | Url | Hugging Face | Description |
|--|--|--|--|--|
comfyui/comfyui_nodes.py
+7 -1
View File
@@ -101,7 +101,13 @@ class LoadCogVideoX_Fun_Model:
if os.path.exists(eas_cache_dir):
model_path = os.path.join(eas_cache_dir, 'CogVideoX_Fun', model)
else:
print(f"Please download cogvideoxfun model to: {model_path}")
model_path = os.path.join(folder_paths.models_dir, "CogVideoX-Fun", model)
if not os.path.exists(model_path):
if os.path.exists(eas_cache_dir):
model_path = os.path.join(eas_cache_dir, 'CogVideoX_Fun', model)
else:
# Detect model is existing or not
print(f"Please download cogvideoxfun model to: {model_path}")
vae = AutoencoderKLCogVideoX.from_pretrained(
model_path,