xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-01-23 12:24:23 +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
set_num_frames(num_frames)
set_num_frames(num_frames) #enhance a video global
# 1. Time embedding
timesteps = timestep

2
model_loading.py
View File

@ -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