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

update diffusers>=0.33.1

Browse Source
This commit is contained in:
sko00o 2025-07-31 14:59:28 +08:00
parent dbc63f622d
commit 389fb0323f
No known key found for this signature in database
GPG Key ID: 240DEE2151D0CF19
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
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
) )

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