From 25d0ede406d3f6d475d5dc110d4080125c9f7ee3 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 22 Dec 2024 18:12:07 +0200
Subject: [PATCH] fix

---
 custom_cogvideox_transformer_3d.py | 2 +-
 model_loading.py                   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/custom_cogvideox_transformer_3d.py b/custom_cogvideox_transformer_3d.py
index efccb33..fb859a5 100644
--- a/custom_cogvideox_transformer_3d.py
+++ b/custom_cogvideox_transformer_3d.py
@@ -555,7 +555,7 @@ class CogVideoXTransformer3DModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
     ):
         batch_size, num_frames, channels, height, width = hidden_states.shape
 
-        set_num_frames(num_frames)
+        set_num_frames(num_frames) #enhance a video global
    
         # 1. Time embedding
         timesteps = timestep
diff --git a/model_loading.py b/model_loading.py
index d8ad72f..da2ce2d 100644
--- a/model_loading.py
+++ b/model_loading.py
@@ -216,7 +216,7 @@ class DownloadAndLoadCogVideoModel:
         download_path = folder_paths.get_folder_paths("CogVideo")[0]
         
         if "Fun" in model:
-            if not "1.1" and not "1.5" in model:
+            if "1.1" not in model and "1.5" not in model:
                 repo_id = "kijai/CogVideoX-Fun-pruned"
                 if "2b" in model:
                     base_path = os.path.join(folder_paths.models_dir, "CogVideoX_Fun", "CogVideoX-Fun-2b-InP") # location of the official model