Upload folder using huggingface_hub

config.json

@@ -4,7 +4,6 @@
   "vae": "vae",
   "lam": "lam",
   "text_encoder": "text_encoder",
-  "crossattn_adapter": "crossattn_adapter",
   "action_processor": "action_processor",
   "transformer": "transformer"
 }

text_encoder/cached_empty_string_embeddings.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:03ab86bd951a0810c5a6768088b47d55a43a8369805d3187f062f2f5e52c3d1b
-size 102762314
+oid sha256:4a2a4eadf8fa63b1a90a538f05557285bebd1687e16805db28d5e911b6acb128
+size 102762335

transformer/config.json

@@ -29,9 +29,9 @@
   "rope_w_extrapolation_ratio": 1.0,
   "shift": 5.0,
   "tie_word_embeddings": false,
-  "timestep_scale": 0.001,
+  "timestep_scale": 1.0,
   "transformers_version": "4.57.3",
   "use_adaln_lora": true,
   "use_crossattn_projection": true,
   "zero_init_action_embedder": false
-}
+}

transformer/model-00003-of-00003.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c12729c1e50e7a659feca865a0364e123606d23de08b876dad94cf3a4ffbbec9
-size 100675824
+oid sha256:5cc38a1ff394798f931d54a3f5fb67fd40168bb95344f78dd0d378c7757f3558
+size 306198928

transformer/model.safetensors.index.json

@@ -3,580 +3,582 @@
     "total_size": 4097425408
   },
   "weight_map": {
-    "x_embedder.proj.1.weight": "model-00001-of-00003.safetensors",
-    "t_embedder.1.linear_1.weight": "model-00001-of-00003.safetensors",
-    "t_embedder.1.linear_2.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.0.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "action_embedder_B_3D.fc1.bias": "model-00002-of-00003.safetensors",
+    "action_embedder_B_3D.fc1.weight": "model-00002-of-00003.safetensors",
+    "action_embedder_B_3D.fc2.bias": "model-00003-of-00003.safetensors",
+    "action_embedder_B_3D.fc2.weight": "model-00003-of-00003.safetensors",
+    "action_embedder_B_D.fc1.bias": "model-00002-of-00003.safetensors",
+    "action_embedder_B_D.fc1.weight": "model-00002-of-00003.safetensors",
+    "action_embedder_B_D.fc2.bias": "model-00002-of-00003.safetensors",
+    "action_embedder_B_D.fc2.weight": "model-00002-of-00003.safetensors",
     "blocks.0.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
     "blocks.0.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
     "blocks.0.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
     "blocks.0.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.1.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
     "blocks.1.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
     "blocks.1.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
     "blocks.1.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
     "blocks.1.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.2.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.3.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.4.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.5.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.6.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.7.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.8.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.9.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.10.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
     "blocks.10.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
     "blocks.10.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
     "blocks.10.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
     "blocks.10.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.11.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.10.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
     "blocks.11.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
     "blocks.11.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
     "blocks.11.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
     "blocks.11.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.12.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.11.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
     "blocks.12.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
     "blocks.12.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
     "blocks.12.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
     "blocks.12.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.mlp.layer1.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.mlp.layer2.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
-    "blocks.13.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.12.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
     "blocks.13.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
     "blocks.13.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
     "blocks.13.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
     "blocks.13.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
-    "blocks.14.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.14.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.14.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.14.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
-    "blocks.14.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
-    "blocks.14.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.14.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.13.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.13.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
     "blocks.14.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.14.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.14.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.14.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.15.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.14.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.14.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.14.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.14.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.14.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
     "blocks.15.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.15.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.15.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.15.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.16.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.15.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.16.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.16.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.16.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.16.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.16.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.17.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.17.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.17.mlp.layer1.weight": "model-00002-of-00003.safetensors",
     "blocks.17.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.17.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.18.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.17.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.18.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.18.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.18.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.18.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.19.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.18.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.19.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.19.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.19.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.19.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.20.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.19.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.2.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.2.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
     "blocks.20.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.20.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.20.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.20.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.21.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.20.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.21.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.21.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.21.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.21.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.22.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.21.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.22.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.22.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.22.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.22.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.23.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.22.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.23.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.23.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.23.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.23.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.24.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.25.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.23.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.24.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.25.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.25.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.25.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.25.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.26.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.25.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.26.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.26.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.26.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.26.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.mlp.layer1.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.mlp.layer2.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
-    "blocks.27.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.26.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
     "blocks.27.adaln_modulation_cross_attn.1.weight": "model-00002-of-00003.safetensors",
     "blocks.27.adaln_modulation_cross_attn.2.weight": "model-00002-of-00003.safetensors",
     "blocks.27.adaln_modulation_mlp.1.weight": "model-00002-of-00003.safetensors",
     "blocks.27.adaln_modulation_mlp.2.weight": "model-00002-of-00003.safetensors",
-    "final_layer.linear.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.adaln_modulation_self_attn.1.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.adaln_modulation_self_attn.2.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.cross_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.cross_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.cross_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.cross_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.cross_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.cross_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.mlp.layer1.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.mlp.layer2.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.self_attn.k_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.self_attn.output_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.self_attn.q_norm.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.27.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "blocks.3.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.3.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.4.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.5.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.6.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.7.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.8.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.adaln_modulation_cross_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.adaln_modulation_cross_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.adaln_modulation_mlp.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.adaln_modulation_mlp.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.adaln_modulation_self_attn.1.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.adaln_modulation_self_attn.2.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.cross_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.cross_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.cross_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.cross_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.cross_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.cross_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.mlp.layer1.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.mlp.layer2.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.self_attn.k_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.self_attn.output_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.self_attn.q_norm.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "blocks.9.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "crossattn_proj.0.bias": "model-00003-of-00003.safetensors",
+    "crossattn_proj.0.weight": "model-00003-of-00003.safetensors",
     "final_layer.adaln_modulation.1.weight": "model-00002-of-00003.safetensors",
     "final_layer.adaln_modulation.2.weight": "model-00002-of-00003.safetensors",
+    "final_layer.linear.weight": "model-00002-of-00003.safetensors",
+    "t_embedder.1.linear_1.weight": "model-00001-of-00003.safetensors",
+    "t_embedder.1.linear_2.weight": "model-00001-of-00003.safetensors",
     "t_embedding_norm.weight": "model-00002-of-00003.safetensors",
-    "action_embedder_B_D.fc1.weight": "model-00002-of-00003.safetensors",
-    "action_embedder_B_D.fc1.bias": "model-00002-of-00003.safetensors",
-    "action_embedder_B_D.fc2.weight": "model-00002-of-00003.safetensors",
-    "action_embedder_B_D.fc2.bias": "model-00002-of-00003.safetensors",
-    "action_embedder_B_3D.fc1.weight": "model-00002-of-00003.safetensors",
-    "action_embedder_B_3D.fc1.bias": "model-00002-of-00003.safetensors",
-    "action_embedder_B_3D.fc2.weight": "model-00003-of-00003.safetensors",
-    "action_embedder_B_3D.fc2.bias": "model-00003-of-00003.safetensors"
+    "x_embedder.proj.1.weight": "model-00001-of-00003.safetensors"
   }
 }