Upload 601 files

transformers/config.json

@@ -0,0 +1,26 @@
+{
+  "_class_name": "MiniMaxH3Transformer3DModel",
+  "_diffusers_version": "0.36.0.dev0",
+  "num_attention_heads": 56,
+  "attention_head_dim": 128,
+  "hidden_size": 5376,
+  "num_layers": 50,
+  "num_refiner_layers": 2,
+  "ffn_dim": 14336,
+  "in_channels": 24,
+  "audio_in_channels": 32,
+  "patch_size": [
+    1,
+    2,
+    2
+  ],
+  "text_dim": 5120,
+  "freq_dim": 256,
+  "time_embed_hidden_dim": 5376,
+  "time_embed_dim": 2688,
+  "rope_freq_dim": 16,
+  "rope_theta": 10000.0,
+  "norm_eps": 1e-05,
+  "qk_norm_eps": 1e-05,
+  "final_norm_eps": 1e-05
+}

transformers/config_nf4.json

@@ -0,0 +1,27 @@
+{
+  "num_attention_heads": 56,
+  "attention_head_dim": 128,
+  "hidden_size": 5376,
+  "num_layers": 50,
+  "num_refiner_layers": 2,
+  "ffn_dim": 14336,
+  "in_channels": 24,
+  "audio_in_channels": 32,
+  "patch_size": [
+    1,
+    2,
+    2
+  ],
+  "text_dim": 5120,
+  "freq_dim": 256,
+  "time_embed_hidden_dim": 5376,
+  "time_embed_dim": 2688,
+  "rope_freq_dim": 16,
+  "rope_theta": 10000.0,
+  "norm_eps": 1e-05,
+  "qk_norm_eps": 1e-05,
+  "final_norm_eps": 1e-05,
+  "_class_name": "MiniMaxH3Transformer3DModel",
+  "_diffusers_version": "0.36.0.dev0",
+  "_name_or_path": "./MiniMax-H3"
+}

transformers/diffusion_pytorch_model.safetensors.index.json

@@ -0,0 +1,645 @@
+{
+  "metadata": {
+    "total_size": 66280430080
+  },
+  "weight_map": {
+    "proj_in.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "proj_in.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "audio_proj_in.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "audio_proj_in.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "context_embedder.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "context_embedder.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "time_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "time_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "time_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "time_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "token_refiner.final_norm.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.norm1.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.norm2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.adaln_proj.linear.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.0.adaln_proj.linear.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.norm1.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.norm2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.adaln_proj.linear.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.1.adaln_proj.linear.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.norm1.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.norm2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.adaln_proj.linear.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.adaln_proj.linear.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.norm1.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.norm2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "norm_out.norm.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "norm_out.linear.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "proj_out.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "audio_proj_out.weight": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "audio_proj_out.bias": "diffusion_pytorch_model-00001-of-00014.safetensors",
+    "transformer_blocks.3.adaln_proj.linear.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.3.adaln_proj.linear.bias": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.norm1.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.norm2.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.adaln_proj.linear.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.4.adaln_proj.linear.bias": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.norm1.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.norm2.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.adaln_proj.linear.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.5.adaln_proj.linear.bias": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.norm1.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.norm2.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.adaln_proj.linear.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.adaln_proj.linear.bias": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.norm1.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.norm2.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00014.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.7.adaln_proj.linear.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.7.adaln_proj.linear.bias": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.norm1.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.norm2.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.adaln_proj.linear.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.8.adaln_proj.linear.bias": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.norm1.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.norm2.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.adaln_proj.linear.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.9.adaln_proj.linear.bias": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.norm1.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.norm2.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.adaln_proj.linear.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.adaln_proj.linear.bias": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.10.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.11.norm1.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.11.norm2.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00014.safetensors",
+    "transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.11.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.11.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.11.adaln_proj.linear.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.11.adaln_proj.linear.bias": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.norm1.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.norm2.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.adaln_proj.linear.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.12.adaln_proj.linear.bias": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.norm1.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.norm2.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.adaln_proj.linear.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.13.adaln_proj.linear.bias": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.norm1.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.norm2.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00014.safetensors",
+    "transformer_blocks.14.adaln_proj.linear.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.14.adaln_proj.linear.bias": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.norm1.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.norm2.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.attn.to_out.0.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.adaln_proj.linear.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.15.adaln_proj.linear.bias": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.norm1.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.norm2.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.attn.to_out.0.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.adaln_proj.linear.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.16.adaln_proj.linear.bias": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.norm1.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.norm2.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.attn.to_out.0.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.adaln_proj.linear.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.17.adaln_proj.linear.bias": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.norm1.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.norm2.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.attn.to_out.0.weight": "diffusion_pytorch_model-00005-of-00014.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.18.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.18.adaln_proj.linear.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.18.adaln_proj.linear.bias": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.norm1.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.norm2.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.attn.to_q.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.attn.to_k.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.attn.to_v.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.attn.to_out.0.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.adaln_proj.linear.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.19.adaln_proj.linear.bias": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.norm1.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.norm2.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.attn.to_q.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.attn.to_k.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.attn.to_v.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.attn.to_out.0.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.adaln_proj.linear.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.20.adaln_proj.linear.bias": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.norm1.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.norm2.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.attn.to_q.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.attn.to_k.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.attn.to_v.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.attn.norm_q.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.attn.norm_k.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.attn.to_out.0.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.adaln_proj.linear.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.21.adaln_proj.linear.bias": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.22.norm1.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.22.norm2.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.22.attn.to_q.weight": "diffusion_pytorch_model-00006-of-00014.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.22.ff.net.2.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.22.adaln_proj.linear.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.22.adaln_proj.linear.bias": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.22.attn.to_k.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.22.attn.to_v.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.22.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.22.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.22.attn.to_out.0.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.norm1.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.norm2.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.attn.to_q.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.attn.to_k.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.attn.to_v.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.attn.to_out.0.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.ff.net.2.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.adaln_proj.linear.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.23.adaln_proj.linear.bias": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.norm1.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.norm2.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.attn.to_q.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.attn.to_k.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.attn.to_v.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.attn.to_out.0.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.ff.net.2.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.adaln_proj.linear.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.24.adaln_proj.linear.bias": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.norm1.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.norm2.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.attn.to_q.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.attn.to_k.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.attn.to_v.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.attn.norm_q.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.attn.norm_k.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.attn.to_out.0.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.ff.net.2.weight": "diffusion_pytorch_model-00007-of-00014.safetensors",
+    "transformer_blocks.25.adaln_proj.linear.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.25.adaln_proj.linear.bias": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.norm1.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.norm2.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.ff.net.2.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.adaln_proj.linear.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.adaln_proj.linear.bias": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.attn.to_q.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.attn.to_k.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.attn.to_v.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.attn.norm_q.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.attn.norm_k.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.26.attn.to_out.0.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.norm1.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.norm2.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.attn.to_q.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.attn.to_k.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.attn.to_v.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.attn.norm_q.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.attn.norm_k.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.attn.to_out.0.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.ff.net.2.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.adaln_proj.linear.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.27.adaln_proj.linear.bias": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.norm1.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.norm2.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.attn.to_q.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.attn.to_k.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.attn.to_v.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.attn.norm_q.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.attn.norm_k.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.attn.to_out.0.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.ff.net.2.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.adaln_proj.linear.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.28.adaln_proj.linear.bias": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.norm1.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.norm2.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.attn.to_q.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.attn.to_k.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.attn.to_v.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.attn.norm_q.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.attn.norm_k.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.attn.to_out.0.weight": "diffusion_pytorch_model-00008-of-00014.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.29.ff.net.2.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.29.adaln_proj.linear.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.29.adaln_proj.linear.bias": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.norm1.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.norm2.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.ff.net.0.proj.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.ff.net.2.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.adaln_proj.linear.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.adaln_proj.linear.bias": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.30.attn.to_out.0.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.norm1.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.norm2.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.attn.to_q.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.attn.to_k.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.attn.to_v.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.attn.norm_q.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.attn.norm_k.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.attn.to_out.0.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.ff.net.0.proj.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.ff.net.2.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.adaln_proj.linear.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.31.adaln_proj.linear.bias": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.norm1.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.norm2.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.attn.to_q.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.attn.to_k.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.attn.to_v.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.attn.norm_q.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.attn.norm_k.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.attn.to_out.0.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.ff.net.0.proj.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.ff.net.2.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.adaln_proj.linear.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.32.adaln_proj.linear.bias": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.33.norm1.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.33.norm2.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.33.attn.to_q.weight": "diffusion_pytorch_model-00009-of-00014.safetensors",
+    "transformer_blocks.33.attn.to_k.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.33.attn.to_v.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.33.attn.norm_q.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.33.attn.norm_k.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.33.attn.to_out.0.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.33.ff.net.2.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.33.adaln_proj.linear.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.33.adaln_proj.linear.bias": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.norm1.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.norm2.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.ff.net.2.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.adaln_proj.linear.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.adaln_proj.linear.bias": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.attn.to_q.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.attn.to_k.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.attn.to_v.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.attn.norm_q.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.attn.norm_k.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.34.attn.to_out.0.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.norm1.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.norm2.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.attn.to_q.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.attn.to_k.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.attn.to_v.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.attn.norm_q.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.attn.norm_k.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.attn.to_out.0.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.ff.net.2.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.adaln_proj.linear.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.35.adaln_proj.linear.bias": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.norm1.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.norm2.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.attn.to_q.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.attn.to_k.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.attn.to_v.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.attn.norm_q.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.attn.norm_k.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.attn.to_out.0.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.ff.net.0.proj.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.ff.net.2.weight": "diffusion_pytorch_model-00010-of-00014.safetensors",
+    "transformer_blocks.36.adaln_proj.linear.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.36.adaln_proj.linear.bias": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.norm1.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.norm2.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.attn.to_q.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.attn.to_k.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.attn.to_v.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.attn.norm_q.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.attn.norm_k.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.attn.to_out.0.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.ff.net.0.proj.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.ff.net.2.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.adaln_proj.linear.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.37.adaln_proj.linear.bias": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.norm1.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.norm2.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.ff.net.0.proj.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.ff.net.2.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.adaln_proj.linear.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.adaln_proj.linear.bias": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.attn.to_q.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.attn.to_k.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.attn.to_v.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.attn.norm_q.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.attn.norm_k.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.38.attn.to_out.0.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.norm1.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.norm2.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.attn.to_q.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.attn.to_k.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.attn.to_v.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.attn.norm_q.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.attn.norm_k.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.attn.to_out.0.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.ff.net.0.proj.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.ff.net.2.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.adaln_proj.linear.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.39.adaln_proj.linear.bias": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.norm1.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.norm2.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.attn.to_q.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.attn.to_k.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.attn.to_v.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.attn.norm_q.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.attn.norm_k.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.attn.to_out.0.weight": "diffusion_pytorch_model-00011-of-00014.safetensors",
+    "transformer_blocks.40.ff.net.0.proj.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.40.ff.net.2.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.40.adaln_proj.linear.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.40.adaln_proj.linear.bias": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.norm1.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.norm2.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.attn.to_q.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.attn.to_k.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.attn.to_v.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.attn.norm_q.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.attn.norm_k.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.attn.to_out.0.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.ff.net.2.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.adaln_proj.linear.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.41.adaln_proj.linear.bias": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.norm1.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.norm2.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.ff.net.2.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.adaln_proj.linear.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.adaln_proj.linear.bias": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.attn.to_q.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.attn.to_k.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.attn.to_v.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.attn.norm_q.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.attn.norm_k.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.42.attn.to_out.0.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.norm1.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.norm2.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.attn.to_q.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.attn.to_k.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.attn.to_v.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.attn.norm_q.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.attn.norm_k.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.attn.to_out.0.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.ff.net.2.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.adaln_proj.linear.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.43.adaln_proj.linear.bias": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.44.norm1.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.44.norm2.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.44.attn.to_q.weight": "diffusion_pytorch_model-00012-of-00014.safetensors",
+    "transformer_blocks.44.attn.to_k.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.44.attn.to_v.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.44.attn.norm_q.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.44.attn.norm_k.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.44.attn.to_out.0.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.44.ff.net.2.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.44.adaln_proj.linear.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.44.adaln_proj.linear.bias": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.norm1.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.norm2.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.attn.to_q.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.attn.to_k.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.attn.to_v.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.attn.norm_q.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.attn.norm_k.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.attn.to_out.0.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.ff.net.0.proj.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.ff.net.2.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.adaln_proj.linear.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.45.adaln_proj.linear.bias": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.norm1.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.norm2.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.ff.net.0.proj.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.ff.net.2.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.adaln_proj.linear.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.adaln_proj.linear.bias": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.attn.to_q.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.attn.to_k.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.attn.to_v.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.attn.norm_q.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.attn.norm_k.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.46.attn.to_out.0.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.norm1.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.norm2.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.attn.to_q.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.attn.to_k.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.attn.to_v.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.attn.norm_q.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.attn.norm_k.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.attn.to_out.0.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.ff.net.0.proj.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "transformer_blocks.47.ff.net.2.weight": "diffusion_pytorch_model-00013-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.norm1.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.norm2.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.norm1.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.norm2.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "token_refiner.refiner_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.47.adaln_proj.linear.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.47.adaln_proj.linear.bias": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.norm1.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.norm2.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.attn.to_q.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.attn.to_k.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.attn.to_v.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.attn.norm_q.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.attn.norm_k.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.attn.to_out.0.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.ff.net.0.proj.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.ff.net.2.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.adaln_proj.linear.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.48.adaln_proj.linear.bias": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.norm1.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.norm2.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.attn.to_q.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.attn.to_k.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.attn.to_v.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.attn.norm_q.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.attn.norm_k.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.attn.to_out.0.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.ff.net.0.proj.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.ff.net.2.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.adaln_proj.linear.weight": "diffusion_pytorch_model-00014-of-00014.safetensors",
+    "transformer_blocks.49.adaln_proj.linear.bias": "diffusion_pytorch_model-00014-of-00014.safetensors"
+  }
+}

transformers/metadata.json

@@ -0,0 +1,14 @@
+{
+  "format": "MiniMax-H3-NF4",
+  "version": 1,
+  "component_name": "transformer",
+  "component_type": "transformer",
+  "quant_type": "nf4",
+  "compute_dtype": "bfloat16",
+  "total_linear": 370,
+  "quantized": 370,
+  "unquantized": 0,
+  "incremental": true,
+  "other": 210,
+  "nonlinear_export": "5b_export_nonlinear_NF4.py"
+}

transformers/weights/audio_proj_in.safetensors

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

transformers/weights/audio_proj_out.safetensors

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

transformers/weights/context_embedder.safetensors

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

transformers/weights/norm_out__linear.safetensors

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

transformers/weights/other-norm_out__norm__weight.safetensors

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

transformers/weights/other-token_refiner__final_norm__weight.safetensors

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

transformers/weights/other-token_refiner__refiner_blocks__0__attn__norm_k__weight.safetensors

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

transformers/weights/other-token_refiner__refiner_blocks__0__attn__norm_q__weight.safetensors

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

transformers/weights/other-token_refiner__refiner_blocks__0__norm1__weight.safetensors

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

transformers/weights/other-token_refiner__refiner_blocks__0__norm2__weight.safetensors

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

transformers/weights/other-token_refiner__refiner_blocks__1__attn__norm_k__weight.safetensors

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

transformers/weights/other-token_refiner__refiner_blocks__1__attn__norm_q__weight.safetensors

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

transformers/weights/other-token_refiner__refiner_blocks__1__norm1__weight.safetensors

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

transformers/weights/other-token_refiner__refiner_blocks__1__norm2__weight.safetensors

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

transformers/weights/other-transformer_blocks__0__attn__norm_k__weight.safetensors

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

transformers/weights/other-transformer_blocks__0__attn__norm_q__weight.safetensors

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

transformers/weights/other-transformer_blocks__0__norm1__weight.safetensors

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

transformers/weights/other-transformer_blocks__0__norm2__weight.safetensors

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

transformers/weights/other-transformer_blocks__10__attn__norm_k__weight.safetensors

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

transformers/weights/other-transformer_blocks__10__attn__norm_q__weight.safetensors

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

transformers/weights/other-transformer_blocks__10__norm1__weight.safetensors

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

transformers/weights/other-transformer_blocks__10__norm2__weight.safetensors

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

transformers/weights/other-transformer_blocks__11__attn__norm_k__weight.safetensors

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

transformers/weights/other-transformer_blocks__11__attn__norm_q__weight.safetensors

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

transformers/weights/other-transformer_blocks__11__norm1__weight.safetensors

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

transformers/weights/other-transformer_blocks__11__norm2__weight.safetensors

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

transformers/weights/other-transformer_blocks__12__attn__norm_k__weight.safetensors

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

transformers/weights/other-transformer_blocks__12__attn__norm_q__weight.safetensors

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

transformers/weights/other-transformer_blocks__12__norm1__weight.safetensors

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

transformers/weights/other-transformer_blocks__12__norm2__weight.safetensors

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

transformers/weights/other-transformer_blocks__13__attn__norm_k__weight.safetensors

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

transformers/weights/other-transformer_blocks__13__attn__norm_q__weight.safetensors

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

transformers/weights/other-transformer_blocks__13__norm1__weight.safetensors

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

transformers/weights/other-transformer_blocks__13__norm2__weight.safetensors

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

transformers/weights/other-transformer_blocks__14__attn__norm_k__weight.safetensors

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

transformers/weights/other-transformer_blocks__14__attn__norm_q__weight.safetensors

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

transformers/weights/other-transformer_blocks__14__norm1__weight.safetensors

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

transformers/weights/other-transformer_blocks__14__norm2__weight.safetensors

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

transformers/weights/other-transformer_blocks__15__attn__norm_k__weight.safetensors

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

transformers/weights/other-transformer_blocks__15__attn__norm_q__weight.safetensors

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

transformers/weights/other-transformer_blocks__15__norm1__weight.safetensors

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

transformers/weights/other-transformer_blocks__15__norm2__weight.safetensors

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

transformers/weights/other-transformer_blocks__16__attn__norm_k__weight.safetensors

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

transformers/weights/other-transformer_blocks__16__attn__norm_q__weight.safetensors

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