chengzeyi commited on
Commit
348da65
·
verified ·
1 Parent(s): bb53e82

Update transformer/config.json

Browse files
Files changed (1) hide show
  1. transformer/config.json +3 -4
transformer/config.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "_class_name": "WanTransformer3DModel",
3
- "_diffusers_version": "0.35.0.dev0",
4
- "added_kv_proj_dim": null,
5
  "attention_head_dim": 128,
6
  "cross_attn_norm": true,
7
  "eps": 1e-06,
8
  "ffn_dim": 13824,
9
  "freq_dim": 256,
10
- "image_dim": null,
11
  "in_channels": 36,
12
  "num_attention_heads": 40,
13
  "num_layers": 40,
@@ -17,7 +17,6 @@
17
  2,
18
  2
19
  ],
20
- "pos_embed_seq_len": null,
21
  "qk_norm": "rms_norm_across_heads",
22
  "rope_max_seq_len": 1024,
23
  "text_dim": 4096
 
1
  {
2
  "_class_name": "WanTransformer3DModel",
3
+ "_diffusers_version": "0.33.0.dev0",
4
+ "added_kv_proj_dim": 5120,
5
  "attention_head_dim": 128,
6
  "cross_attn_norm": true,
7
  "eps": 1e-06,
8
  "ffn_dim": 13824,
9
  "freq_dim": 256,
10
+ "image_dim": 1280,
11
  "in_channels": 36,
12
  "num_attention_heads": 40,
13
  "num_layers": 40,
 
17
  2,
18
  2
19
  ],
 
20
  "qk_norm": "rms_norm_across_heads",
21
  "rope_max_seq_len": 1024,
22
  "text_dim": 4096