We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7ffecbc commit b4e11e7Copy full SHA for b4e11e7
src/diffusers/models/transformers/transformer_cogview4.py
@@ -36,7 +36,7 @@ class CogView4PatchEmbed(nn.Module):
36
def __init__(
37
self,
38
in_channels: int = 16,
39
- hidden_size: int = 4096,
+ hidden_size: int = 2560,
40
patch_size: int = 2,
41
text_hidden_size: int = 4096,
42
):
0 commit comments