diff --git "a/diffusion_pytorch_model.safetensors.index.json" "b/diffusion_pytorch_model.safetensors.index.json"
new file mode 100644--- /dev/null
+++ "b/diffusion_pytorch_model.safetensors.index.json"
@@ -0,0 +1,1564 @@
+{
+  "metadata": {
+    "total_size": 16630894807
+  },
+  "weight_map": {
+    "norm_out.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "norm_out.linear.weight.absmax": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "norm_out.linear.weight.quant_map": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "norm_out.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "patch_embed.proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "patch_embed.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "pos_frequencies": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "proj_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "proj_out.weight.absmax": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "proj_out.weight.quant_map": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "proj_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "time_embed.caption_proj.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.caption_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.caption_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.caption_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.caption_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_kv.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_kv.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_kv.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_kv.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_kv.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_q.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.pooler.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_1.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_1.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_1.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "time_embed.timestep_embedder.linear_2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.0.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.1.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.10.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.11.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.12.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.13.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.14.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.15.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.16.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.17.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.18.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.19.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.2.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.20.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.21.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.22.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.23.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.24.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.25.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.26.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.27.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.28.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_k_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_k_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_k_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_q_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_q_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_q_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_v_proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_v_proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.add_v_proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_add_out.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_add_out.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_add_out.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_k.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_k.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_k.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_q.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_q.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_q.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_v.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_v.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.attn1.to_v.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff_context.net.0.proj.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff_context.net.0.proj.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff_context.net.0.proj.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff_context.net.2.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff_context.net.2.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.ff_context.net.2.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1_context.linear.weight.absmax": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1_context.linear.weight.quant_map": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.29.norm1_context.linear.weight.quant_state.bitsandbytes__nf4": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.3.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.30.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.31.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.31.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.31.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.31.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.32.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.32.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.33.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.34.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.35.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.36.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.37.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.38.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.39.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.4.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.4.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.40.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.40.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.41.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.42.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.43.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.44.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.45.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.to_add_out.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.46.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.norm1_context.linear_1.bias": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.47.norm1_context.linear_1.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "transformer_blocks.5.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.5.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.6.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.7.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.8.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.to_add_out.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.to_add_out.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00002.safetensors",
+    "transformer_blocks.9.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00002.safetensors"
+  }
+}