diff --git a/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/config.json b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..4e5185528fdf802a785024cbebc61f29fdfa0fd0
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/config.json
@@ -0,0 +1,20 @@
+{
+  "_class_name": "FluxTransformer2DModel",
+  "_diffusers_version": "0.39.0.dev0",
+  "_name_or_path": "/home/work/MMSearch/cwf/search/model/FLUX.1-Fill-dev",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    16,
+    56,
+    56
+  ],
+  "guidance_embeds": true,
+  "in_channels": 384,
+  "joint_attention_dim": 4096,
+  "num_attention_heads": 24,
+  "num_layers": 19,
+  "num_single_layers": 38,
+  "out_channels": 64,
+  "patch_size": 1,
+  "pooled_projection_dim": 768
+}
diff --git a/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..d2125816da262ab4faaf4410327ef868f0d2f558
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:afd89aaf519f185dfa17f77f7c28581a03ed5c16313f015efce4299b0fb9732f
+size 9985006384
diff --git a/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..cb7ce65020f1831beeb4b6fb88f6f4b096b637c9
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca73df9ac9ed12a757fbb02171dbedd6b5554c56b8542aa89592f682ed235d28
+size 9949328904
diff --git a/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..6834d6c6dabe4864d4b6015974db2f974e507cbe
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61e52c8529790ec7636b76809de32cceb39740d268b3fb1923526563da4523b7
+size 3870584832
diff --git a/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model.safetensors.index.json b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model.safetensors.index.json
new file mode 100644
index 0000000000000000000000000000000000000000..21f6c277a5b661e77f9b259c05ca7dad2a29f559
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-2000/transformer/diffusion_pytorch_model.safetensors.index.json
@@ -0,0 +1,1167 @@
+{
+  "metadata": {
+    "total_size": 23804782720
+  },
+  "weight_map": {
+    "context_embedder.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "context_embedder.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "norm_out.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "x_embedder.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "x_embedder.weight": "diffusion_pytorch_model-00001-of-00003.safetensors"
+  }
+}
diff --git a/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/config.json b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..4e5185528fdf802a785024cbebc61f29fdfa0fd0
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/config.json
@@ -0,0 +1,20 @@
+{
+  "_class_name": "FluxTransformer2DModel",
+  "_diffusers_version": "0.39.0.dev0",
+  "_name_or_path": "/home/work/MMSearch/cwf/search/model/FLUX.1-Fill-dev",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    16,
+    56,
+    56
+  ],
+  "guidance_embeds": true,
+  "in_channels": 384,
+  "joint_attention_dim": 4096,
+  "num_attention_heads": 24,
+  "num_layers": 19,
+  "num_single_layers": 38,
+  "out_channels": 64,
+  "patch_size": 1,
+  "pooled_projection_dim": 768
+}
diff --git a/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..75ec5ba38cde2a0bb0cbce5c04e5dfc0abf5f221
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:15e29505a93324365f8bfd31942f908d7695990865ffa41fff54bdfb98d3f878
+size 9985006384
diff --git a/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..ec3df95b3cef8751cb033b5869dc511b3fb11119
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc36bd4dc8b6ca04449f343cce4d5c28ed744e68e9d035845d9bc56083e44ba1
+size 9949328904
diff --git a/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..185548228f6103b77656bab8315b45b4d177fb65
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f869ed618acb85a42d3e738d040fe0e62c79bf4ac63a08864a3a93489122a8b8
+size 3870584832
diff --git a/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model.safetensors.index.json b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model.safetensors.index.json
new file mode 100644
index 0000000000000000000000000000000000000000..21f6c277a5b661e77f9b259c05ca7dad2a29f559
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-4000/transformer/diffusion_pytorch_model.safetensors.index.json
@@ -0,0 +1,1167 @@
+{
+  "metadata": {
+    "total_size": 23804782720
+  },
+  "weight_map": {
+    "context_embedder.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "context_embedder.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "norm_out.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "x_embedder.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "x_embedder.weight": "diffusion_pytorch_model-00001-of-00003.safetensors"
+  }
+}
diff --git a/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/config.json b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..4e5185528fdf802a785024cbebc61f29fdfa0fd0
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/config.json
@@ -0,0 +1,20 @@
+{
+  "_class_name": "FluxTransformer2DModel",
+  "_diffusers_version": "0.39.0.dev0",
+  "_name_or_path": "/home/work/MMSearch/cwf/search/model/FLUX.1-Fill-dev",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    16,
+    56,
+    56
+  ],
+  "guidance_embeds": true,
+  "in_channels": 384,
+  "joint_attention_dim": 4096,
+  "num_attention_heads": 24,
+  "num_layers": 19,
+  "num_single_layers": 38,
+  "out_channels": 64,
+  "patch_size": 1,
+  "pooled_projection_dim": 768
+}
diff --git a/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..6db0fb2b1bef3fc56cb0f2ca0471c771bff5aa1a
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00001-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1171721b20f670e0012a0d56a5853cb4941f52b0959b6753e79567d3eb3e453
+size 9985006384
diff --git a/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e3e889cf9ac06b9369735300568d2e4444033557
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00002-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c63ce16604587bf6a0003e13563e6bc8170befbe581cb2020e88119924418383
+size 9949328904
diff --git a/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e40908526f3c49d7d1ef19951367861b53a42c1e
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model-00003-of-00003.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:84599df76e4b9029bf5894acd027451c72dad026f21eea39bfa30ec975a33fee
+size 3870584832
diff --git a/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model.safetensors.index.json b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model.safetensors.index.json
new file mode 100644
index 0000000000000000000000000000000000000000..21f6c277a5b661e77f9b259c05ca7dad2a29f559
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/checkpoint-6000/transformer/diffusion_pytorch_model.safetensors.index.json
@@ -0,0 +1,1167 @@
+{
+  "metadata": {
+    "total_size": 23804782720
+  },
+  "weight_map": {
+    "context_embedder.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "context_embedder.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "norm_out.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "norm_out.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.0.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.1.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.10.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.11.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.12.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.13.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.14.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.15.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.16.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.17.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.18.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.19.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.2.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.20.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.21.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.22.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.23.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.24.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.25.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.26.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.27.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.28.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.29.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.3.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.30.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.31.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.32.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.33.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.34.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.35.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.36.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.norm.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.norm.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_mlp.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_mlp.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.37.proj_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.4.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.5.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.6.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.7.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.8.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.norm.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.norm.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_mlp.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_mlp.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "single_transformer_blocks.9.proj_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.guidance_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.text_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "time_text_embed.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.0.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.1.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.10.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.11.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.12.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.13.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.14.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.14.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.15.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.16.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.17.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.ff_context.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1_context.linear.bias": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.18.norm1_context.linear.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.2.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.3.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.4.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.5.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.6.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.7.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.8.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_add_out.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_add_out.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_k.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_out.0.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_q.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_v.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.ff_context.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1_context.linear.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "transformer_blocks.9.norm1_context.linear.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "x_embedder.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
+    "x_embedder.weight": "diffusion_pytorch_model-00001-of-00003.safetensors"
+  }
+}
diff --git a/trained-full-multiview-caption-stage1/transformer/config.json b/trained-full-multiview-caption-stage1/transformer/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..4e5185528fdf802a785024cbebc61f29fdfa0fd0
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/transformer/config.json
@@ -0,0 +1,20 @@
+{
+  "_class_name": "FluxTransformer2DModel",
+  "_diffusers_version": "0.39.0.dev0",
+  "_name_or_path": "/home/work/MMSearch/cwf/search/model/FLUX.1-Fill-dev",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    16,
+    56,
+    56
+  ],
+  "guidance_embeds": true,
+  "in_channels": 384,
+  "joint_attention_dim": 4096,
+  "num_attention_heads": 24,
+  "num_layers": 19,
+  "num_single_layers": 38,
+  "out_channels": 64,
+  "patch_size": 1,
+  "pooled_projection_dim": 768
+}
diff --git a/trained-full-multiview-caption-stage1/transformer/diffusion_pytorch_model.safetensors b/trained-full-multiview-caption-stage1/transformer/diffusion_pytorch_model.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..e78114bb2fb7cb378573c0fffd1ba6f42b493527
--- /dev/null
+++ b/trained-full-multiview-caption-stage1/transformer/diffusion_pytorch_model.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d4aef5e98e5a137ab140c48b1a307183c182f6be57fdead446eff301af5287da
+size 23804921696