Upload magi_human variant: sr_1080p

sr_1080p/model_index.json

@@ -0,0 +1,16 @@
+{
+  "_class_name": "MagiHumanSR1080pPipeline",
+  "_diffusers_version": "0.33.0",
+  "transformer": [
+    "diffusers",
+    "MagiHumanDiT"
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowUniPCMultistepScheduler"
+  ],
+  "sr_transformer": [
+    "diffusers",
+    "MagiHumanDiT"
+  ]
+}

sr_1080p/scheduler/scheduler_config.json

@@ -0,0 +1,13 @@
+{
+  "_class_name": "FlowUniPCMultistepScheduler",
+  "_diffusers_version": "0.33.0",
+  "num_train_timesteps": 1000,
+  "solver_order": 2,
+  "prediction_type": "flow_prediction",
+  "shift": 5.0,
+  "predict_x0": true,
+  "solver_type": "bh2",
+  "lower_order_final": true,
+  "disable_corrector": [],
+  "flow_shift": 5.0
+}

sr_1080p/sr_transformer/config.json

@@ -0,0 +1,40 @@
+{
+  "_class_name": "MagiHumanDiT",
+  "_diffusers_version": "0.33.0",
+  "num_layers": 40,
+  "hidden_size": 5120,
+  "head_dim": 128,
+  "num_query_groups": 8,
+  "video_in_channels": 192,
+  "audio_in_channels": 64,
+  "text_in_channels": 3584,
+  "mm_layers": [
+    0,
+    1,
+    2,
+    3,
+    36,
+    37,
+    38,
+    39
+  ],
+  "local_attn_layers": [],
+  "gelu7_layers": [
+    0,
+    1,
+    2,
+    3
+  ],
+  "post_norm_layers": [],
+  "enable_attn_gating": true,
+  "activation_type": "swiglu7",
+  "patch_size": [
+    1,
+    2,
+    2
+  ],
+  "spatial_rope_interpolation": "extra",
+  "tread_selection_rate": 0.5,
+  "tread_start_layer_idx": 2,
+  "tread_end_layer_idx": 25
+}

sr_1080p/sr_transformer/diffusion_pytorch_model-00001-of-00007.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:acba3bc07eb2ed803c3fdd09dabbdf6a81badd26257f493d3ebf0fd3bb769bfe
+size 5093802208

sr_1080p/sr_transformer/diffusion_pytorch_model-00002-of-00007.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:66fad159696ff27d9266040095e28930b60bd6ba724f4cd5cd66208280f1dc0b
+size 5220491016

sr_1080p/sr_transformer/diffusion_pytorch_model-00003-of-00007.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ee744c679fcb20509adea1018004199ebca8d25c228f3a9f6e66d68a98c6ff4e
+size 4964896496

sr_1080p/sr_transformer/diffusion_pytorch_model-00004-of-00007.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42a07a7ba585c339280e60a7462177155f902f7a166ab712de2198edef582b5c
+size 5144492296

sr_1080p/sr_transformer/diffusion_pytorch_model-00005-of-00007.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:57d5f63d6e4131bfd590bb450337e7accb0a69effe800a0185f8c8bccfb287a9
+size 4814086536

sr_1080p/sr_transformer/diffusion_pytorch_model-00006-of-00007.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:62ef5edb7ad70086133ee1150fd85aeb4caea2a853c3bb8e05e5c2c988dae74b
+size 4565497360

sr_1080p/sr_transformer/diffusion_pytorch_model-00007-of-00007.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:98764dc3165a48338892d543a5f353dbfaf1df5fd0a1d342c0a416103e34cdba
+size 838779000

sr_1080p/sr_transformer/diffusion_pytorch_model.safetensors.index.json

@@ -0,0 +1,338 @@
+{
+  "metadata": {
+    "total_size": 30642008128
+  },
+  "weight_map": {
+    "adapter.audio_embedder.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.0.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.1.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.1.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.1.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.1.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.11.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.12.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.12.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.12.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.14.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.15.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.15.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.15.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.17.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.17.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.19.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.3.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.3.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.3.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.3.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.8.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.9.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.9.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "adapter.text_embedder.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.19.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.21.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.22.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.22.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.22.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.22.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.22.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.22.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.25.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.27.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.28.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.28.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.28.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.28.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.28.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.28.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.31.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.33.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.34.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.34.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.34.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.34.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.34.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.34.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.37.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.37.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.38.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.38.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.0.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.12.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.12.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.12.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "block.layers.12.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.15.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.15.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.15.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.15.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.18.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.2.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.2.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.2.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.5.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.5.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.5.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.5.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.7.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.9.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.9.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.9.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.9.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "adapter.audio_embedder.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.20.attention.linear_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.21.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.21.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.21.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.21.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.21.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.21.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.24.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.26.attention.linear_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.27.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.27.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.27.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.27.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.27.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.27.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.30.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.32.attention.linear_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.33.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.33.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.33.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.33.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.33.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.33.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.36.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.36.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.36.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.36.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "final_linear_audio.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "final_linear_video.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "final_norm_audio.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "final_norm_video.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.0.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.11.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.11.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.11.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "block.layers.11.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.14.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.16.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.17.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.17.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.17.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.17.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.17.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.3.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.3.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.8.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.8.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.8.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.8.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.17.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.20.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.22.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.23.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.23.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.23.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.23.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.23.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.23.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.26.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.28.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.29.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.29.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.29.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.29.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.29.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.29.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.32.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.34.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.35.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.35.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.35.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.35.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.35.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.35.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.37.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.39.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "adapter.rope.bands": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "adapter.text_embedder.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.0.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.0.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.0.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.1.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "block.layers.1.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.3.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.39.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.4.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.4.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.4.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.4.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.4.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.4.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.6.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.6.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.6.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.10.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.10.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.10.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.10.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.10.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.10.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.13.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.13.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.13.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.13.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.16.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.18.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.19.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.19.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.19.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.19.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.19.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.19.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.22.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.24.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.25.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.25.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.25.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.25.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.25.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.25.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.28.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.30.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.31.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.31.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.31.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.31.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.31.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.6.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.9.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.9.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "adapter.video_embedder.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.0.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.0.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.3.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.31.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.34.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.36.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.37.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.38.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.38.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.39.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.39.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.39.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.39.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.39.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "block.layers.39.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.4.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.4.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.5.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.6.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.10.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.10.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.12.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.13.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.13.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.13.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.15.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.16.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.16.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.16.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.16.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.16.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.16.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.20.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.20.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.20.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.20.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.20.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.20.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.23.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.25.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.26.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.26.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.26.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.26.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.26.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.26.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.29.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.31.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.32.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.6.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.6.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.6.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.8.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.8.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.8.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "adapter.video_embedder.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.1.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.1.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.2.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.2.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.2.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.2.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.32.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.32.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.32.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.32.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.32.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.35.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.36.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.38.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.38.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.38.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "block.layers.38.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.11.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.11.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.11.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.13.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.14.attention.k_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.14.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.14.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.14.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.14.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.14.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.18.attention.k_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.18.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.18.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.18.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.18.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.18.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.2.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.21.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.23.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.24.attention.k_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.24.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.24.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.24.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.24.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.5.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.5.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.5.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.7.attention.k_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.7.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.7.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.7.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.7.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.7.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.7.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.24.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.27.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.29.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.30.attention.k_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.30.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.30.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.30.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.30.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.30.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.33.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.35.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.36.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.36.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.37.attention.k_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.37.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.37.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "block.layers.37.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors"
+  }
+}

sr_1080p/transformer/config.json

@@ -0,0 +1,40 @@
+{
+  "_class_name": "MagiHumanDiT",
+  "_diffusers_version": "0.33.0",
+  "num_layers": 40,
+  "hidden_size": 5120,
+  "head_dim": 128,
+  "num_query_groups": 8,
+  "video_in_channels": 192,
+  "audio_in_channels": 64,
+  "text_in_channels": 3584,
+  "mm_layers": [
+    0,
+    1,
+    2,
+    3,
+    36,
+    37,
+    38,
+    39
+  ],
+  "local_attn_layers": [],
+  "gelu7_layers": [
+    0,
+    1,
+    2,
+    3
+  ],
+  "post_norm_layers": [],
+  "enable_attn_gating": true,
+  "activation_type": "swiglu7",
+  "patch_size": [
+    1,
+    2,
+    2
+  ],
+  "spatial_rope_interpolation": "extra",
+  "tread_selection_rate": 0.5,
+  "tread_start_layer_idx": 2,
+  "tread_end_layer_idx": 25
+}

sr_1080p/transformer/diffusion_pytorch_model-00001-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:91cca957655320140e73850a96b4f32616a9e19fdd9f1afa3790ab15abf64d6e
+size 5359513152

sr_1080p/transformer/diffusion_pytorch_model-00002-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bba999eb9dde204e1f42223ad6e21455955abb6626ae0eb085761ca8daee1b1e
+size 4845236248

sr_1080p/transformer/diffusion_pytorch_model-00003-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:146beb17b24e456b6f364e0e6d0ac293b8d92ac13245ba78232e2ac42170a467
+size 4999797120

sr_1080p/transformer/diffusion_pytorch_model-00004-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1296e1cc73efc07c2806dfa4247cb823fed72c80dd54444b607b4403862c74e
+size 5344542680

sr_1080p/transformer/diffusion_pytorch_model-00005-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a9c0e0d4044a17c9835a0e2ebe458a79ef51be4e076d91aa8ad4db0ba22643c1
+size 5347139480

sr_1080p/transformer/diffusion_pytorch_model-00006-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9df676362b8c73d0f61eba29df9024d00b5f5bf5ebd30237b6806ae0cd87cb7c
+size 4745816328

sr_1080p/transformer/diffusion_pytorch_model.safetensors.index.json

@@ -0,0 +1,338 @@
+{
+  "metadata": {
+    "total_size": 30642008128
+  },
+  "weight_map": {
+    "adapter.audio_embedder.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "adapter.text_embedder.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "adapter.text_embedder.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.0.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.1.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.12.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.17.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.21.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.24.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.25.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.25.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.27.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.29.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.32.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.35.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.38.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.39.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.39.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.4.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.7.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.9.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.9.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.9.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "final_linear_video.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.10.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.14.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.15.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.16.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.17.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.2.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.21.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.22.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.25.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.27.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.29.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.31.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.31.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.31.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.35.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.36.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.36.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.38.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.4.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.5.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.6.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "adapter.video_embedder.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.11.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.12.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.12.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.13.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.15.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.16.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.18.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.19.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.22.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.22.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.22.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.27.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.28.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.28.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.30.attention.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.31.attention.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.32.mlp.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.5.attention.pre_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "final_norm_video.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.0.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.10.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.11.attention.linear_qkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.12.attention.linear_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.17.mlp.down_proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "block.layers.19.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.2.attention.linear_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.20.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.20.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.20.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.21.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.23.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.24.attention.linear_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.25.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.32.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.34.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.34.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.36.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.37.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.37.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.37.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.4.attention.linear_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.6.mlp.down_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.8.attention.linear_qkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.0.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.10.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.10.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.11.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.11.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.12.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.13.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.14.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.2.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.2.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.21.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.21.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.23.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.26.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.26.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.27.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.27.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.28.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.29.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.3.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.3.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.31.attention.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.31.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.34.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.34.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.36.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.36.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.6.mlp.pre_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.7.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.7.attention.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.8.attention.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "final_norm_audio.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.0.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "block.layers.1.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.10.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.16.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.16.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.16.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.17.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.21.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.24.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.27.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.28.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.28.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.29.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.3.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.35.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.37.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.39.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.5.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.6.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.7.attention.linear_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.7.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.8.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.9.attention.linear_qkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.9.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "adapter.audio_embedder.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.0.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.1.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.1.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.13.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.15.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.16.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.18.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.19.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.19.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.2.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.20.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.23.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.24.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.25.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.28.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.29.attention.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.30.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.30.mlp.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.33.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.34.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.35.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.39.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.4.attention.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.8.attention.pre_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.0.mlp.down_proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "block.layers.1.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.10.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.11.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.12.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.13.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.13.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.13.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.18.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.2.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.2.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.23.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.26.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.3.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.30.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.33.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.33.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.34.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.35.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.4.mlp.down_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.5.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.7.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.9.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.9.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "adapter.video_embedder.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.14.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.15.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.15.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.17.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.17.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.19.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.20.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.20.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.21.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.22.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.23.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.24.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.24.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.25.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.26.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.3.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.30.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.33.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.33.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.34.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.35.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.35.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.36.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.37.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.37.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.37.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.38.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.38.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.39.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.4.attention.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.5.attention.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.5.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.6.attention.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.8.mlp.pre_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.1.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.11.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.12.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.12.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.14.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.15.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.15.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.18.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.18.attention.linear_qkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.19.attention.linear_proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "block.layers.19.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.20.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.21.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.22.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.22.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.26.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.26.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.26.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.27.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.3.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.30.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.30.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.32.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.33.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.35.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.38.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.38.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.4.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.6.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.8.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.8.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "adapter.rope.bands": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.0.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.10.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.14.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.14.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.16.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.17.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.18.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.18.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.2.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.23.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.3.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.32.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.32.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.33.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.36.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.37.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.38.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.4.mlp.pre_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.5.attention.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.6.attention.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.7.attention.pre_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "final_linear_audio.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.1.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.11.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.13.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.14.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.15.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.17.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.18.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.19.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.22.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.23.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.24.attention.linear_qkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.24.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.25.attention.linear_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.25.mlp.down_proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "block.layers.27.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.28.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.29.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.29.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.31.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.32.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.33.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.34.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.38.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.39.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.39.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.39.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.5.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.7.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.1.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.10.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.11.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.13.attention.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.16.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.20.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.26.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.29.attention.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.31.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.6.attention.pre_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.8.attention.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.9.mlp.pre_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.0.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.14.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.23.attention.linear_qkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.28.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.3.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.30.mlp.up_gate_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.32.attention.linear_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "block.layers.36.mlp.down_proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors"
+  }
+}