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

Merge pull request #391 from huixingyun/main

Browse Source 
Update minimal diffusers version
This commit is contained in:
Jukka Seppänen 2025-08-07 14:03:37 +03:00 committed by GitHub
commit fdb8abd279
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
embeddings.py
View File

@ -67,8 +67,9 @@ class CogVideoXPatchEmbed(nn.Module):
post_time_compression_frames, post_time_compression_frames,
self.spatial_interpolation_scale, self.spatial_interpolation_scale,
self.temporal_interpolation_scale, self.temporal_interpolation_scale,
output_type="pt",
) )
pos_embedding = torch.from_numpy(pos_embedding).flatten(0, 1) pos_embedding = pos_embedding.flatten(0, 1)
joint_pos_embedding = torch.zeros( joint_pos_embedding = torch.zeros(
1, self.max_text_seq_length + num_patches, self.embed_dim, requires_grad=False 1, self.max_text_seq_length + num_patches, self.embed_dim, requires_grad=False
) )
@ -173,6 +174,8 @@ def get_3d_rotary_pos_embed(
grid_t = np.arange(temporal_size, dtype=np.float32) grid_t = np.arange(temporal_size, dtype=np.float32)
grid_t = np.linspace(0, temporal_size, temporal_size, endpoint=False, dtype=np.float32) grid_t = np.linspace(0, temporal_size, temporal_size, endpoint=False, dtype=np.float32)
elif grid_type == "slice": elif grid_type == "slice":
if max_size is None:
raise ValueError("`max_size` must be provided when `grid_type` is 'slice'")
max_h, max_w = max_size max_h, max_w = max_size
grid_size_h, grid_size_w = grid_size grid_size_h, grid_size_w = grid_size
grid_h = np.arange(max_h, dtype=np.float32) grid_h = np.arange(max_h, dtype=np.float32)

2
requirements.txt
View File

@ -1,5 +1,5 @@
huggingface_hub huggingface_hub
diffusers>=0.31.0 diffusers>=0.33.1
accelerate>=0.33.0 accelerate>=0.33.0
einops einops
peft peft