diff --git "a/transformer/diffusion_pytorch_model.safetensors.index.json" "b/transformer/diffusion_pytorch_model.safetensors.index.json"
new file mode 100644--- /dev/null
+++ "b/transformer/diffusion_pytorch_model.safetensors.index.json"
@@ -0,0 +1,1808 @@
+{
+  "metadata": {
+    "total_size": 33306432640
+  },
+  "weight_map": {
+    "cond_type_embed.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.proj_in.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.proj_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.time_text_embed.text_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.time_text_embed.text_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.time_text_embed.text_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.time_text_embed.text_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.time_text_embed.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.time_text_embed.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.time_text_embed.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.time_text_embed.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.norm1.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.norm1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.norm2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.norm2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.norm_out.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.0.norm_out.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.norm1.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.norm1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.norm2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.norm2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.norm_out.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder.token_refiner.refiner_blocks.1.norm_out.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder_2.linear_1.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder_2.linear_1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder_2.linear_2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder_2.linear_2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder_2.linear_3.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder_2.linear_3.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder_2.norm.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "context_embedder_2.norm.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "image_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "image_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "image_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "image_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "image_embedder.norm_in.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "image_embedder.norm_in.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "image_embedder.norm_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "image_embedder.norm_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "norm_out.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "proj_out.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "time_embed.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "time_embed.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "time_embed.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "time_embed.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.0.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.1.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.10.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.11.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.12.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.13.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.14.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.15.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.15.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.16.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.16.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.17.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.18.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.19.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.2.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.20.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.20.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.21.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.22.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.23.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.24.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.25.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.26.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.27.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.28.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.29.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.3.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.30.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.30.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.31.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00004.safetensors",
+    "transformer_blocks.32.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.32.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.33.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.34.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.35.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.36.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.37.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.38.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.39.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.4.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.40.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.40.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.41.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.42.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.43.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.44.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.45.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.46.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.ff_context.net.2.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.ff_context.net.2.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.47.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.attn.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_add_out.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_add_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.ff_context.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.ff_context.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.48.norm1.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.norm1.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.norm1_context.linear.bias": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.48.norm1_context.linear.weight": "diffusion_pytorch_model-00003-of-00004.safetensors",
+    "transformer_blocks.49.attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.add_q_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.add_q_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_add_out.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_add_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.ff_context.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.ff_context.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.norm1.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.norm1.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.norm1_context.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.49.norm1_context.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.5.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.50.attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.add_q_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.add_q_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_add_out.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_add_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.ff_context.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.ff_context.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.norm1.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.norm1.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.norm1_context.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.50.norm1_context.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.add_q_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.add_q_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_add_out.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_add_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.ff_context.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.ff_context.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.norm1.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.norm1.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.norm1_context.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.51.norm1_context.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.add_q_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.add_q_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_add_out.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_add_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.ff_context.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.ff_context.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.norm1.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.norm1.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.norm1_context.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.52.norm1_context.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.add_q_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.add_q_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.norm_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.norm_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_add_out.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_add_out.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_k.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_k.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_out.0.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_out.0.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_q.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_q.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_v.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.attn.to_v.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.ff_context.net.2.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.ff_context.net.2.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.norm1.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.norm1.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.norm1_context.linear.bias": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.53.norm1_context.linear.weight": "diffusion_pytorch_model-00004-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.6.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.7.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.8.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "transformer_blocks.9.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "x_embedder.proj.bias": "diffusion_pytorch_model-00001-of-00004.safetensors",
+    "x_embedder.proj.weight": "diffusion_pytorch_model-00001-of-00004.safetensors"
+  }
+}