diff --git a/image_encoder/config.json b/image_encoder/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..4679ff7351920ccdde01d38516195504cffc235b
--- /dev/null
+++ b/image_encoder/config.json
@@ -0,0 +1,22 @@
+{
+  "architectures": [
+    "CLIPVisionModel"
+  ],
+  "attention_dropout": 0.0,
+  "dropout": 0.0,
+  "hidden_act": "gelu",
+  "hidden_size": 1280,
+  "image_size": 224,
+  "initializer_factor": 1.0,
+  "initializer_range": 0.02,
+  "intermediate_size": 5120,
+  "layer_norm_eps": 1e-05,
+  "model_type": "clip_vision_model",
+  "num_attention_heads": 16,
+  "num_channels": 3,
+  "num_hidden_layers": 32,
+  "patch_size": 14,
+  "projection_dim": 1024,
+  "torch_dtype": "float32",
+  "transformers_version": "4.49.0"
+}
diff --git a/image_encoder/model.safetensors b/image_encoder/model.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..9a1b4670d88773c628ee27ea508c8a0cf8bc98fa
--- /dev/null
+++ b/image_encoder/model.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8eb46f477ef5e1859b659014aed6ca56cdc207c12cb7a0f9d61b4d80a1a7bb84
+size 2523128312
diff --git a/image_processor/preprocessor_config.json b/image_processor/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..ae07b4c27c8e8d50505ee232210ef38d1e1cb2a2
--- /dev/null
+++ b/image_processor/preprocessor_config.json
@@ -0,0 +1,28 @@
+{
+  "crop_size": {
+    "height": 224,
+    "width": 224
+  },
+  "do_center_crop": false,
+  "do_convert_rgb": true,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.48145466,
+    0.4578275,
+    0.40821073
+  ],
+  "image_processor_type": "CLIPImageProcessor",
+  "image_std": [
+    0.26862954,
+    0.26130258,
+    0.27577711
+  ],
+  "resample": 3,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}
diff --git a/model_index.json b/model_index.json
new file mode 100644
index 0000000000000000000000000000000000000000..a645980af634c02009a28904a29e869ffed406ca
--- /dev/null
+++ b/model_index.json
@@ -0,0 +1,32 @@
+{
+  "_class_name": "WanI2VPipeline",
+  "_diffusers_version": "0.33.0.dev0",
+  "image_encoder": [
+    "transformers",
+    "CLIPVisionModel"
+  ],
+  "image_processor": [
+    "transformers",
+    "CLIPImageProcessor"
+  ],
+  "scheduler": [
+    "diffusers",
+    "UniPCMultistepScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "UMT5EncoderModel"
+  ],
+  "tokenizer": [
+    "transformers",
+    "T5TokenizerFast"
+  ],
+  "transformer": [
+    "diffusers",
+    "WanTransformer3DModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLWan"
+  ]
+}
diff --git a/scheduler/scheduler_config.json b/scheduler/scheduler_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..347f0c7955d00ca19ef775d6504e2d77f7b191d1
--- /dev/null
+++ b/scheduler/scheduler_config.json
@@ -0,0 +1,28 @@
+{
+  "_class_name": "UniPCMultistepScheduler",
+  "_diffusers_version": "0.33.0.dev0",
+  "beta_end": 0.02,
+  "beta_schedule": "linear",
+  "beta_start": 0.0001,
+  "disable_corrector": [],
+  "dynamic_thresholding_ratio": 0.995,
+  "final_sigmas_type": "zero",
+  "flow_shift": 7.0,
+  "lower_order_final": true,
+  "num_train_timesteps": 1000,
+  "predict_x0": true,
+  "prediction_type": "flow_prediction",
+  "rescale_betas_zero_snr": false,
+  "sample_max_value": 1.0,
+  "solver_order": 2,
+  "solver_p": null,
+  "solver_type": "bh2",
+  "steps_offset": 0,
+  "thresholding": false,
+  "timestep_spacing": "linspace",
+  "trained_betas": null,
+  "use_beta_sigmas": false,
+  "use_exponential_sigmas": false,
+  "use_flow_sigmas": true,
+  "use_karras_sigmas": false
+}
diff --git a/transformer/config.json b/transformer/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..9fc17d3d03509600c2c9fac211fbd1c8fb6483a0
--- /dev/null
+++ b/transformer/config.json
@@ -0,0 +1,26 @@
+{
+  "_class_name": "WanTransformer3DModel",
+  "_diffusers_version": "0.33.0.dev0",
+  "add_img_emb": true,
+  "added_kv_proj_dim": 5120,
+  "attention_head_dim": 128,
+  "cross_attn_norm": true,
+  "eps": 1e-06,
+  "ffn_dim": 13824,
+  "freq_dim": 256,
+  "in_channels": 36,
+  "num_attention_heads": 40,
+  "num_layers": 40,
+  "out_channels": 16,
+  "patch_size": [
+    1,
+    2,
+    2
+  ],
+  "qk_norm": true,
+  "text_dim": 4096,
+  "window_size": [
+    -1,
+    -1
+  ]
+}
diff --git a/transformer/diffusion_pytorch_model-00001-of-00007.safetensors b/transformer/diffusion_pytorch_model-00001-of-00007.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..d4d3e3f276f6cfb9477bc0113a6fa0a02fd61507
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00001-of-00007.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:457639497b204e838c0b5e7f5955e8b1b0f9f04213bd9853e40cd77771569685
+size 9952163512
diff --git a/transformer/diffusion_pytorch_model-00002-of-00007.safetensors b/transformer/diffusion_pytorch_model-00002-of-00007.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..a4e203ba13d5e6dcd4e699a4afae744e5f6f0a1a
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00002-of-00007.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eda5a7f06db0164b852b52fc56db9cb82c502e963f0e7d407af8e13bac31826b
+size 9797226656
diff --git a/transformer/diffusion_pytorch_model-00003-of-00007.safetensors b/transformer/diffusion_pytorch_model-00003-of-00007.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..bc67d063c8aad221fd71cfae695d30802e15efae
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00003-of-00007.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:be4dfe5a12ac35c857fe307e91899d0f9c473551c2a828e80718f340489b27bd
+size 9975437232
diff --git a/transformer/diffusion_pytorch_model-00004-of-00007.safetensors b/transformer/diffusion_pytorch_model-00004-of-00007.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..dbd5dd1d069a35a246ca0c05eba6d7a1c3dc078e
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00004-of-00007.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:744a014df498e5b579d34a12e9ea836b2cd9adf9e0ef77b4f7378ad762091573
+size 9975566544
diff --git a/transformer/diffusion_pytorch_model-00005-of-00007.safetensors b/transformer/diffusion_pytorch_model-00005-of-00007.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..010a01ac9336c3d62d53edb6c0926ed6989af6b2
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00005-of-00007.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b49e16fed3592ffc07d65503f39517f3190d8e52130418706ea1ba678f207050
+size 9902022768
diff --git a/transformer/diffusion_pytorch_model-00006-of-00007.safetensors b/transformer/diffusion_pytorch_model-00006-of-00007.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..b4fc8fca93ede0e5428261f8677173043bc20525
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00006-of-00007.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d173bad73ebcdde9fe4efef487d1df9c993dfc3590b041e05e7806418479be52
+size 9902063944
diff --git a/transformer/diffusion_pytorch_model-00007-of-00007.safetensors b/transformer/diffusion_pytorch_model-00007-of-00007.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..c2c34660abe1ba1493fb6e75bd8f28efb009a0a3
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00007-of-00007.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:68b4173328f39217e177fe40cf94e509a213955aea842b086f3fd17e6c286832
+size 6075990120
diff --git a/transformer/diffusion_pytorch_model.safetensors.index.json b/transformer/diffusion_pytorch_model.safetensors.index.json
new file mode 100644
index 0000000000000000000000000000000000000000..ee0d46aa79c6eca88e2424c33681d52ed5858855
--- /dev/null
+++ b/transformer/diffusion_pytorch_model.safetensors.index.json
@@ -0,0 +1,1350 @@
+{
+  "metadata": {
+    "total_size": 65580334336
+  },
+  "weight_map": {
+    "blocks.0.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.ffn.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.ffn.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.ffn.2.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.ffn.2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.modulation": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.norm3.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.0.norm3.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.ffn.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.ffn.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.ffn.2.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.ffn.2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.modulation": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.norm3.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.1.norm3.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.10.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.ffn.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.ffn.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.ffn.2.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.ffn.2.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.modulation": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.norm3.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.10.norm3.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.ffn.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.11.ffn.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.11.ffn.2.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.11.ffn.2.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.11.modulation": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.norm3.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.11.norm3.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.12.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.ffn.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.ffn.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.ffn.2.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.ffn.2.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.modulation": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.norm3.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.12.norm3.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.ffn.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.ffn.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.ffn.2.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.ffn.2.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.modulation": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.norm3.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.13.norm3.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.ffn.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.ffn.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.ffn.2.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.ffn.2.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.modulation": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.norm3.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.14.norm3.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.ffn.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.ffn.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.ffn.2.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.ffn.2.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.modulation": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.norm3.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.15.norm3.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.ffn.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.ffn.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.ffn.2.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.ffn.2.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.modulation": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.norm3.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.16.norm3.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.ffn.0.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.ffn.0.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.ffn.2.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.17.ffn.2.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.17.modulation": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.norm3.bias": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.17.norm3.weight": "diffusion_pytorch_model-00003-of-00007.safetensors",
+    "blocks.18.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.ffn.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.ffn.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.ffn.2.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.ffn.2.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.modulation": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.norm3.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.18.norm3.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.ffn.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.ffn.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.ffn.2.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.ffn.2.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.modulation": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.norm3.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.19.norm3.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.2.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.ffn.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.ffn.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.ffn.2.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.ffn.2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.modulation": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.norm3.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.2.norm3.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.20.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.ffn.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.ffn.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.ffn.2.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.ffn.2.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.modulation": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.norm3.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.20.norm3.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.ffn.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.ffn.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.ffn.2.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.ffn.2.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.modulation": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.norm3.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.21.norm3.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.ffn.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.ffn.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.ffn.2.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.ffn.2.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.modulation": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.norm3.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.22.norm3.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.add_k_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.add_k_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.add_v_proj.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.add_v_proj.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.norm_added_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.norm_added_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.ffn.0.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.ffn.0.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.ffn.2.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.ffn.2.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.modulation": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.norm3.bias": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.23.norm3.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.24.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.24.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.24.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.add_k_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.add_k_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.add_v_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.add_v_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.norm_added_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.norm_added_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.ffn.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.ffn.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.ffn.2.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.ffn.2.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.modulation": "diffusion_pytorch_model-00004-of-00007.safetensors",
+    "blocks.24.norm3.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.24.norm3.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.add_k_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.add_k_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.add_v_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.add_v_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.norm_added_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.norm_added_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.ffn.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.ffn.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.ffn.2.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.ffn.2.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.modulation": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.norm3.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.25.norm3.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.add_k_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.add_k_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.add_v_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.add_v_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.norm_added_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.norm_added_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.ffn.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.ffn.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.ffn.2.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.ffn.2.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.modulation": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.norm3.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.26.norm3.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.add_k_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.add_k_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.add_v_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.add_v_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.norm_added_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.norm_added_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.ffn.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.ffn.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.ffn.2.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.ffn.2.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.modulation": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.norm3.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.27.norm3.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.add_k_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.add_k_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.add_v_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.add_v_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.norm_added_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.norm_added_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.ffn.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.ffn.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.ffn.2.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.ffn.2.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.modulation": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.norm3.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.28.norm3.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.add_k_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.add_k_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.add_v_proj.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.add_v_proj.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.norm_added_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.norm_added_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.ffn.0.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.ffn.0.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.ffn.2.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.ffn.2.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.modulation": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.norm3.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.29.norm3.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.3.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.ffn.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.ffn.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.ffn.2.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.ffn.2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.modulation": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.norm3.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.3.norm3.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.30.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.30.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.30.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.30.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.30.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.30.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.30.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.add_k_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.add_k_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.add_v_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.add_v_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.norm_added_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.norm_added_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.ffn.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.ffn.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.ffn.2.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.ffn.2.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.modulation": "diffusion_pytorch_model-00005-of-00007.safetensors",
+    "blocks.30.norm3.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.30.norm3.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.add_k_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.add_k_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.add_v_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.add_v_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.norm_added_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.norm_added_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.ffn.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.ffn.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.ffn.2.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.ffn.2.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.modulation": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.norm3.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.31.norm3.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.add_k_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.add_k_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.add_v_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.add_v_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.norm_added_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.norm_added_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.ffn.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.ffn.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.ffn.2.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.ffn.2.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.modulation": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.norm3.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.32.norm3.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.add_k_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.add_k_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.add_v_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.add_v_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.norm_added_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.norm_added_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.ffn.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.ffn.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.ffn.2.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.ffn.2.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.modulation": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.norm3.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.33.norm3.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.add_k_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.add_k_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.add_v_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.add_v_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.norm_added_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.norm_added_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.ffn.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.ffn.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.ffn.2.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.ffn.2.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.modulation": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.norm3.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.34.norm3.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.add_k_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.add_k_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.add_v_proj.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.add_v_proj.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.norm_added_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.norm_added_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.ffn.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.ffn.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.ffn.2.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.ffn.2.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.modulation": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.norm3.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.35.norm3.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn2.add_k_proj.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.add_k_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.add_v_proj.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.add_v_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.norm_added_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.norm_added_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.attn2.to_k.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.to_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.to_out.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.to_out.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.to_q.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.to_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.to_v.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.attn2.to_v.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.ffn.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.ffn.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.ffn.2.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.ffn.2.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.modulation": "diffusion_pytorch_model-00006-of-00007.safetensors",
+    "blocks.36.norm3.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.36.norm3.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.to_k.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.to_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.to_out.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.to_out.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.to_q.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.to_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.to_v.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn1.to_v.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.add_k_proj.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.add_k_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.add_v_proj.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.add_v_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.norm_added_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.norm_added_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.to_k.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.to_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.to_out.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.to_out.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.to_q.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.to_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.to_v.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.attn2.to_v.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.ffn.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.ffn.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.ffn.2.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.ffn.2.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.modulation": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.norm3.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.37.norm3.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.to_k.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.to_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.to_out.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.to_out.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.to_q.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.to_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.to_v.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn1.to_v.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.add_k_proj.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.add_k_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.add_v_proj.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.add_v_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.norm_added_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.norm_added_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.to_k.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.to_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.to_out.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.to_out.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.to_q.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.to_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.to_v.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.attn2.to_v.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.ffn.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.ffn.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.ffn.2.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.ffn.2.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.modulation": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.norm3.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.38.norm3.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.to_k.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.to_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.to_out.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.to_out.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.to_q.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.to_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.to_v.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn1.to_v.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.add_k_proj.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.add_k_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.add_v_proj.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.add_v_proj.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.norm_added_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.norm_added_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.norm_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.norm_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.to_k.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.to_k.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.to_out.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.to_out.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.to_q.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.to_q.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.to_v.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.attn2.to_v.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.ffn.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.ffn.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.ffn.2.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.ffn.2.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.modulation": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.norm3.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.39.norm3.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "blocks.4.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.norm_added_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.norm_added_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.ffn.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.ffn.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.ffn.2.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.ffn.2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.modulation": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.norm3.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.4.norm3.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.add_v_proj.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.add_v_proj.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.attn2.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.ffn.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.ffn.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.ffn.2.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.ffn.2.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.modulation": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "blocks.5.norm3.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.5.norm3.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.ffn.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.ffn.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.ffn.2.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.ffn.2.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.modulation": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.norm3.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.6.norm3.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.ffn.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.ffn.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.ffn.2.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.ffn.2.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.modulation": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.norm3.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.7.norm3.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.ffn.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.ffn.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.ffn.2.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.ffn.2.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.modulation": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.norm3.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.8.norm3.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.ffn.0.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.ffn.0.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.ffn.2.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.ffn.2.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.modulation": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.norm3.bias": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "blocks.9.norm3.weight": "diffusion_pytorch_model-00002-of-00007.safetensors",
+    "head.head.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "head.head.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "head.modulation": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "img_emb.proj.0.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "img_emb.proj.0.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "img_emb.proj.1.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "img_emb.proj.1.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "img_emb.proj.3.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "img_emb.proj.3.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "img_emb.proj.4.bias": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "img_emb.proj.4.weight": "diffusion_pytorch_model-00007-of-00007.safetensors",
+    "patch_embedding.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "patch_embedding.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "text_embedding.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "text_embedding.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "text_embedding.2.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "text_embedding.2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_embedding.0.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_embedding.0.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_embedding.2.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_embedding.2.weight": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_projection.1.bias": "diffusion_pytorch_model-00001-of-00007.safetensors",
+    "time_projection.1.weight": "diffusion_pytorch_model-00001-of-00007.safetensors"
+  }
+}