YiYiXu HF Staff commited on
Commit
8543b2c
·
verified ·
1 Parent(s): 36b1852

Add diffusers-format transformer weights

Browse files

Adds the diffusers-format Wan-Animate-2 weights for huggingface/diffusers#14413:
- `transformer/`: converted from the research state dict (conversion verified bitwise)
- `scheduler/scheduler_config.json`: the scheduler the release samples with (the current file says `UniPCMultistepScheduler`, which does not match `model_index.json`)
- `modular_model_index.json`: enables `ModularPipeline.from_pretrained` on this repo; the `transformer`/`scheduler` specs pin this PR's revision so it is runnable before merging — they can be set to `null` once merged

transformer/config.json CHANGED
@@ -7,7 +7,6 @@
7
  "ffn_dim": 13824,
8
  "freq_dim": 256,
9
  "in_dim": 36,
10
- "log_scale": -1.3,
11
  "num_heads": 40,
12
  "num_layers": 40,
13
  "out_dim": 16,
@@ -16,17 +15,11 @@
16
  2,
17
  2
18
  ],
19
- "qk_norm": true,
20
  "refer_offset_h": 0,
21
  "refer_offset_t": 1,
22
  "refer_offset_w": -1,
23
  "refer_stride": 1,
24
- "sparse_type": 0,
25
  "text_dim": 4096,
26
  "text_len": 512,
27
- "use_img_emb": true,
28
- "window_size": [
29
- -1,
30
- -1
31
- ]
32
  }
 
7
  "ffn_dim": 13824,
8
  "freq_dim": 256,
9
  "in_dim": 36,
 
10
  "num_heads": 40,
11
  "num_layers": 40,
12
  "out_dim": 16,
 
15
  2,
16
  2
17
  ],
 
18
  "refer_offset_h": 0,
19
  "refer_offset_t": 1,
20
  "refer_offset_w": -1,
21
  "refer_stride": 1,
 
22
  "text_dim": 4096,
23
  "text_len": 512,
24
+ "use_img_emb": true
 
 
 
 
25
  }
transformer/diffusion_pytorch_model-00001-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:83c9f8fe04ae01429c3e46053d1e55e2134fd1942df63ff3fe7b7aa7d2aeaca4
3
- size 9874594352
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebc182727d948cb61cac929af3b2887e0a4b7c8b683b7356628cf0350d332c0a
3
+ size 9874593104
transformer/diffusion_pytorch_model-00002-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a222285e4eb588d37315c3ccc155a2d33c91c2e93fc7de428005c944ff268f9c
3
- size 9975380624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e808cf7a7e7aaf358478a2198152f9e3ec8609c66712be397c21b4b7ec7ef501
3
+ size 9975379320
transformer/diffusion_pytorch_model-00003-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0eea9449bfc81f553f804bf2ddd08abeb3ceb043008e3f364b94e1ac5149858c
3
- size 9954402920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:444eeea631fe6d2df80780f521f6c7716bd079fc177279ba09d2d20479f2dfaf
3
+ size 9954401600
transformer/diffusion_pytorch_model-00004-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:eda459bcd5fe68a5a6fea6e758fe0f70466d31c3f53d54fc443aca876884a2d2
3
- size 2985521896
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:130fd0a278a06096e483f08b2492ddd86b89fef8676b3d7b30556b3230a9b005
3
+ size 2985521520
transformer/diffusion_pytorch_model.safetensors.index.json CHANGED
The diff for this file is too large to render. See raw diff