diff --git a/models_huggingface.yaml b/models_huggingface.yaml
index b7b92bc..f1af797 100644
--- a/models_huggingface.yaml
+++ b/models_huggingface.yaml
@@ -121,8 +121,12 @@ model_categories:
       resolution: 720p
       notes: State-of-the-art text-to-video generation, auto-downloaded by DownloadAndLoadCogVideoModel node
       files:
-        - source: "transformer/diffusion_pytorch_model.safetensors"
-          dest: "cogvideox-5b-transformer.safetensors"
+        - source: "transformer/diffusion_pytorch_model-00001-of-00002.safetensors"
+          dest: "cogvideox-5b-transformer-00001-of-00002.safetensors"
+        - source: "transformer/diffusion_pytorch_model-00002-of-00002.safetensors"
+          dest: "cogvideox-5b-transformer-00002-of-00002.safetensors"
+        - source: "transformer/diffusion_pytorch_model.safetensors.index.json"
+          dest: "cogvideox-5b-transformer.safetensors.index.json"
 
     - repo_id: THUDM/CogVideoX-5b-I2V
       description: CogVideoX-5B-I2V - Image-to-video generation
@@ -136,8 +140,14 @@ model_categories:
       resolution: 720p
       notes: Image-to-video model, auto-downloaded by DownloadAndLoadCogVideoModel node
       files:
-        - source: "transformer/diffusion_pytorch_model.safetensors"
-          dest: "cogvideox-5b-i2v-transformer.safetensors"
+        - source: "transformer/diffusion_pytorch_model-00001-of-00003.safetensors"
+          dest: "cogvideox-5b-i2v-transformer-00001-of-00003.safetensors"
+        - source: "transformer/diffusion_pytorch_model-00002-of-00003.safetensors"
+          dest: "cogvideox-5b-i2v-transformer-00002-of-00003.safetensors"
+        - source: "transformer/diffusion_pytorch_model-00003-of-00003.safetensors"
+          dest: "cogvideox-5b-i2v-transformer-00003-of-00003.safetensors"
+        - source: "transformer/diffusion_pytorch_model.safetensors.index.json"
+          dest: "cogvideox-5b-i2v-transformer.safetensors.index.json"
 
     - repo_id: stabilityai/stable-video-diffusion-img2vid
       description: SVD - 14 frame image-to-video