xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-09 21:04:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
kijai 2024-12-22 18:12:07 +02:00
parent f16d38a5d2
commit 25d0ede406
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
custom_cogvideox_transformer_3d.py
View File

@ -555,7 +555,7 @@ class CogVideoXTransformer3DModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
): ):
batch_size, num_frames, channels, height, width = hidden_states.shape 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 # 1. Time embedding
timesteps = timestep timesteps = timestep

2
model_loading.py
View File

@ -216,7 +216,7 @@ class DownloadAndLoadCogVideoModel:
download_path = folder_paths.get_folder_paths("CogVideo")[0] download_path = folder_paths.get_folder_paths("CogVideo")[0]
if "Fun" in model: 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" repo_id = "kijai/CogVideoX-Fun-pruned"
if "2b" in model: if "2b" in model:
base_path = os.path.join(folder_paths.models_dir, "CogVideoX_Fun", "CogVideoX-Fun-2b-InP") # location of the official model base_path = os.path.join(folder_paths.models_dir, "CogVideoX_Fun", "CogVideoX-Fun-2b-InP") # location of the official model