Skip to content

Commit 3849cae

Browse files
authored
Update src/diffusers/models/transformers/cogvideox_transformer_3d.py
1 parent f054c44 commit 3849cae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/cogvideox_transformer_3d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ class CogVideoXTransformer3DModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
171171
time_embed_dim (`int`, defaults to `512`):
172172
Output dimension of timestep embeddings.
173173
ofs_embed_dim (`int`, defaults to `512`):
174-
scaling factor in the VAE process for the Image-to-Video (I2V) transformation in CogVideoX1.5-5B.
174+
Output dimension of "ofs" embeddings used in CogVideoX-5b-I2B in version 1.5
175175
text_embed_dim (`int`, defaults to `4096`):
176176
Input dimension of text embeddings from the text encoder.
177177
num_layers (`int`, defaults to `30`):

0 commit comments

Comments
 (0)