diff --git "a/transformer/diffusion_pytorch_model.safetensors.index.json" "b/transformer/diffusion_pytorch_model.safetensors.index.json"
--- "a/transformer/diffusion_pytorch_model.safetensors.index.json"
+++ "b/transformer/diffusion_pytorch_model.safetensors.index.json"
@@ -1,1299 +1,388 @@
 {
   "metadata": {
-    "total_size": 32789757568
+    "total_size": 2985508480
   },
   "weight_map": {
-    "blocks.0.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.0.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.1.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.10.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.11.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.11.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.11.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.11.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.11.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.12.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.12.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.13.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.14.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.15.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.16.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.17.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.18.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.19.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.2.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.2.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.20.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.20.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.21.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.22.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.o.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.o.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.23.block.self_attn.v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.24.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
-    "blocks.24.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.24.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.25.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.26.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.27.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.28.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.29.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.3.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.3.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.30.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.30.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.31.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.32.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.33.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.34.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.35.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.cross_attn.k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.o.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.o.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.cross_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.cross_attn.v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.ffn.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.ffn.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.ffn.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.ffn.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.36.block.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.o.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.o.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.36.block.self_attn.v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
-    "blocks.37.block.cross_attn.k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.o.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.o.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.ffn.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.ffn.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.ffn.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.ffn.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.modulation": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.norm3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.norm3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.o.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.o.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.37.block.self_attn.v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.o.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.o.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.ffn.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.ffn.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.ffn.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.ffn.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.modulation": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.norm3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.norm3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.o.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.o.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.38.block.self_attn.v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.o.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.o.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.ffn.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.ffn.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.ffn.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.ffn.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.modulation": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.norm3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.norm3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.o.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.o.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.39.block.self_attn.v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "blocks.4.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.4.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.5.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.6.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.7.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.8.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.k_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.norm_k_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.v_img.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.cross_attn.v_img.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.o.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.o.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "blocks.9.block.self_attn.v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "head.head.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "head.head.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "head.modulation": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "img_emb.proj.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "img_emb.proj.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "img_emb.proj.1.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "img_emb.proj.1.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "img_emb.proj.3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "img_emb.proj.3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "img_emb.proj.4.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
-    "img_emb.proj.4.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.0.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.0.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.1.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.10.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.2.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.3.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.4.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.5.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.6.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.7.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.8.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.ffn.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.ffn.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.ffn.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.ffn.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.modulation": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.norm3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.norm3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.9.self_attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
     "patch_embedding.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
     "patch_embedding.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
     "text_embedding.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
@@ -1305,6 +394,917 @@
     "time_embedding.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
     "time_embedding.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
     "time_projection.1.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
-    "time_projection.1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors"
+    "time_projection.1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "blocks.11.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.11.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.11.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.11.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.12.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.13.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.14.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.15.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.16.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.17.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.18.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.19.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.20.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.21.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.22.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.ffn.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.ffn.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.ffn.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.ffn.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.norm3.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.norm3.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.23.self_attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.24.modulation": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "blocks.24.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.24.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.25.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.26.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.27.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.28.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.29.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.30.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.31.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.32.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.33.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.34.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.ffn.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.ffn.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.ffn.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.ffn.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.35.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.cross_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.cross_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.modulation": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.norm3.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.norm3.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.self_attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "blocks.36.cross_attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.ffn.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.ffn.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.ffn.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.36.ffn.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.ffn.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.ffn.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.ffn.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.ffn.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.modulation": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.norm3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.norm3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.37.self_attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.ffn.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.ffn.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.ffn.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.ffn.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.modulation": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.norm3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.norm3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.38.self_attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.cross_attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.ffn.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.ffn.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.ffn.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.ffn.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.modulation": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.norm3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.norm3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "blocks.39.self_attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "head.head.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "head.head.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "head.modulation": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "img_emb.proj.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "img_emb.proj.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "img_emb.proj.1.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "img_emb.proj.1.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "img_emb.proj.3.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "img_emb.proj.3.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "img_emb.proj.4.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "img_emb.proj.4.weight": "diffusion_pytorch_model-00004-of-00004.safetensors"
   }
 }