xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-16 17:51:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'kijai:main' into main

Browse Source
This commit is contained in:
zhilemann 2024-12-23 04:17:18 +03:00 committed by GitHub
commit 4a07eee589
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -639,7 +639,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