fix

2024-12-22 18:12:07 +02:00
parent f16d38a5d2
commit 25d0ede406
2 changed files with 2 additions and 2 deletions
@@ -555,7 +555,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