bubbliiiing commited on
Commit
8c31827
·
1 Parent(s): 24d60ce

Create Weights

Browse files
Files changed (2) hide show
  1. config.json +21 -7
  2. diffusion_pytorch_model.safetensors +3 -0
config.json CHANGED
@@ -1,14 +1,28 @@
1
  {
2
- "_class_name": "WanModel",
3
- "_diffusers_version": "0.30.0",
4
- "dim": 1536,
 
5
  "eps": 1e-06,
6
- "ffn_dim": 8960,
7
  "freq_dim": 256,
 
 
8
  "in_dim": 36,
9
  "model_type": "i2v",
10
- "num_heads": 12,
11
- "num_layers": 30,
12
  "out_dim": 16,
13
- "text_len": 512
 
 
 
 
 
 
 
 
 
 
 
14
  }
 
1
  {
2
+ "_class_name": "WanTransformer3DModel",
3
+ "_diffusers_version": "0.31.0",
4
+ "cross_attn_norm": true,
5
+ "dim": 5120,
6
  "eps": 1e-06,
7
+ "ffn_dim": 13824,
8
  "freq_dim": 256,
9
+ "hidden_size": 2048,
10
+ "in_channels": 36,
11
  "in_dim": 36,
12
  "model_type": "i2v",
13
+ "num_heads": 40,
14
+ "num_layers": 40,
15
  "out_dim": 16,
16
+ "patch_size": [
17
+ 1,
18
+ 2,
19
+ 2
20
+ ],
21
+ "qk_norm": true,
22
+ "text_dim": 4096,
23
+ "text_len": 512,
24
+ "window_size": [
25
+ -1,
26
+ -1
27
+ ]
28
  }
diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b51cef1f388f234e14fe4a9b7b60e71c8c614aea200a0764daa86c845d52243
3
+ size 32789894056