diff --git a/comfyui/README.md b/comfyui/README.md
index f63f404..d632be5 100644
--- a/comfyui/README.md
+++ b/comfyui/README.md
@@ -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 |
 |--|--|--|--|--|
diff --git a/comfyui/comfyui_nodes.py b/comfyui/comfyui_nodes.py
index 30c57ae..cc21102 100644
--- a/comfyui/comfyui_nodes.py
+++ b/comfyui/comfyui_nodes.py
@@ -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,