diff --git a/.gitattributes b/.gitattributes
index da4fda9973e248828e1a8ffb45585d30782f7c6d..479ac2ca4ac7830797455c6c968452ffa2824ce2 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -47,3 +47,5 @@ media/ltx-video_i2v_example_00007.gif filter=lfs diff=lfs merge=lfs -text
 media/ltx-video_i2v_example_00008.gif filter=lfs diff=lfs merge=lfs -text
 media/ltx-video_i2v_example_00009.gif filter=lfs diff=lfs merge=lfs -text
 media/trailer.gif filter=lfs diff=lfs merge=lfs -text
+*.sft filter=lfs diff=lfs merge=lfs -text
+*.gif filter=lfs diff=lfs merge=lfs -text
diff --git a/model_index.json b/model_index.json
new file mode 100644
index 0000000000000000000000000000000000000000..d928724e82854203c895ee5fc956da465ffe6103
--- /dev/null
+++ b/model_index.json
@@ -0,0 +1,24 @@
+{
+  "_class_name": "LTXConditionPipeline",
+  "_diffusers_version": "0.34.0.dev0",
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "T5EncoderModel"
+  ],
+  "tokenizer": [
+    "transformers",
+    "T5Tokenizer"
+  ],
+  "transformer": [
+    "diffusers",
+    "LTXVideoTransformer3DModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLLTXVideo"
+  ]
+}
diff --git a/scheduler/scheduler_config.json b/scheduler/scheduler_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..d100002d36531d0895d6b738044725815e5ea06c
--- /dev/null
+++ b/scheduler/scheduler_config.json
@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.34.0.dev0",
+  "base_image_seq_len": 256,
+  "base_shift": 0.5,
+  "invert_sigmas": false,
+  "max_image_seq_len": 4096,
+  "max_shift": 1.15,
+  "num_train_timesteps": 1000,
+  "shift": 1.0,
+  "shift_terminal": null,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": false,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}
diff --git a/text_encoder/config.json b/text_encoder/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..d32c5a60ffb46d93202500fd3bea1f8a4b9ff0e7
--- /dev/null
+++ b/text_encoder/config.json
@@ -0,0 +1,31 @@
+{
+  "architectures": [
+    "T5EncoderModel"
+  ],
+  "classifier_dropout": 0.0,
+  "d_ff": 10240,
+  "d_kv": 64,
+  "d_model": 4096,
+  "decoder_start_token_id": 0,
+  "dense_act_fn": "gelu_new",
+  "dropout_rate": 0.1,
+  "eos_token_id": 1,
+  "feed_forward_proj": "gated-gelu",
+  "initializer_factor": 1.0,
+  "is_encoder_decoder": true,
+  "is_gated_act": true,
+  "layer_norm_epsilon": 1e-06,
+  "model_type": "t5",
+  "num_decoder_layers": 24,
+  "num_heads": 64,
+  "num_layers": 24,
+  "output_past": true,
+  "pad_token_id": 0,
+  "relative_attention_max_distance": 128,
+  "relative_attention_num_buckets": 32,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.51.3",
+  "use_cache": true,
+  "vocab_size": 32128
+}
diff --git a/text_encoder/model-00001-of-00004.safetensors b/text_encoder/model-00001-of-00004.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..18c5aabb9b2b13497a1d695dca7c6aa53010ab12
--- /dev/null
+++ b/text_encoder/model-00001-of-00004.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a68b2c8c080696a10109612a649bc69330991ecfea65930ccfdfbdb011f2686
+size 4989319680
diff --git a/text_encoder/model-00002-of-00004.safetensors b/text_encoder/model-00002-of-00004.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..99a5b982f77018015431bcac90f77fb50ef2485b
--- /dev/null
+++ b/text_encoder/model-00002-of-00004.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8ed6556d7507e38af5b428c605fb2a6f2bdb7e80bd481308b865f7a40c551ca
+size 4999830656
diff --git a/text_encoder/model-00003-of-00004.safetensors b/text_encoder/model-00003-of-00004.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..03e07005d6976043a9bbe738c61e39a60ebabaca
--- /dev/null
+++ b/text_encoder/model-00003-of-00004.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c831635f83041f83faf0024b39c6ecb21b45d70dd38a63ea5bac6c7c6e5e558c
+size 4865612720
diff --git a/text_encoder/model-00004-of-00004.safetensors b/text_encoder/model-00004-of-00004.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..2c6513c8a7cfd4f59f71837c19fb28bfc5aae25e
--- /dev/null
+++ b/text_encoder/model-00004-of-00004.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:02a5f2d69205be92ad48fe5d712d38c2ff55627969116aeffc58bd75a28da468
+size 4194506688
diff --git a/text_encoder/model.safetensors.index.json b/text_encoder/model.safetensors.index.json
new file mode 100644
index 0000000000000000000000000000000000000000..0900b8ebd70052d4cccc2eacf6e07d75bb1f9cb8
--- /dev/null
+++ b/text_encoder/model.safetensors.index.json
@@ -0,0 +1,226 @@
+{
+  "metadata": {
+    "total_size": 19049242624
+  },
+  "weight_map": {
+    "encoder.block.0.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.relative_attention_bias.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.12.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.18.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.5.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.final_layer_norm.weight": "model-00004-of-00004.safetensors",
+    "shared.weight": "model-00001-of-00004.safetensors"
+  }
+}
diff --git a/tokenizer/added_tokens.json b/tokenizer/added_tokens.json
new file mode 100644
index 0000000000000000000000000000000000000000..3f5132007c4fcf42b75b65c8b6aa49c7098bcdf4
--- /dev/null
+++ b/tokenizer/added_tokens.json
@@ -0,0 +1,102 @@
+{
+  "<extra_id_0>": 32099,
+  "<extra_id_10>": 32089,
+  "<extra_id_11>": 32088,
+  "<extra_id_12>": 32087,
+  "<extra_id_13>": 32086,
+  "<extra_id_14>": 32085,
+  "<extra_id_15>": 32084,
+  "<extra_id_16>": 32083,
+  "<extra_id_17>": 32082,
+  "<extra_id_18>": 32081,
+  "<extra_id_19>": 32080,
+  "<extra_id_1>": 32098,
+  "<extra_id_20>": 32079,
+  "<extra_id_21>": 32078,
+  "<extra_id_22>": 32077,
+  "<extra_id_23>": 32076,
+  "<extra_id_24>": 32075,
+  "<extra_id_25>": 32074,
+  "<extra_id_26>": 32073,
+  "<extra_id_27>": 32072,
+  "<extra_id_28>": 32071,
+  "<extra_id_29>": 32070,
+  "<extra_id_2>": 32097,
+  "<extra_id_30>": 32069,
+  "<extra_id_31>": 32068,
+  "<extra_id_32>": 32067,
+  "<extra_id_33>": 32066,
+  "<extra_id_34>": 32065,
+  "<extra_id_35>": 32064,
+  "<extra_id_36>": 32063,
+  "<extra_id_37>": 32062,
+  "<extra_id_38>": 32061,
+  "<extra_id_39>": 32060,
+  "<extra_id_3>": 32096,
+  "<extra_id_40>": 32059,
+  "<extra_id_41>": 32058,
+  "<extra_id_42>": 32057,
+  "<extra_id_43>": 32056,
+  "<extra_id_44>": 32055,
+  "<extra_id_45>": 32054,
+  "<extra_id_46>": 32053,
+  "<extra_id_47>": 32052,
+  "<extra_id_48>": 32051,
+  "<extra_id_49>": 32050,
+  "<extra_id_4>": 32095,
+  "<extra_id_50>": 32049,
+  "<extra_id_51>": 32048,
+  "<extra_id_52>": 32047,
+  "<extra_id_53>": 32046,
+  "<extra_id_54>": 32045,
+  "<extra_id_55>": 32044,
+  "<extra_id_56>": 32043,
+  "<extra_id_57>": 32042,
+  "<extra_id_58>": 32041,
+  "<extra_id_59>": 32040,
+  "<extra_id_5>": 32094,
+  "<extra_id_60>": 32039,
+  "<extra_id_61>": 32038,
+  "<extra_id_62>": 32037,
+  "<extra_id_63>": 32036,
+  "<extra_id_64>": 32035,
+  "<extra_id_65>": 32034,
+  "<extra_id_66>": 32033,
+  "<extra_id_67>": 32032,
+  "<extra_id_68>": 32031,
+  "<extra_id_69>": 32030,
+  "<extra_id_6>": 32093,
+  "<extra_id_70>": 32029,
+  "<extra_id_71>": 32028,
+  "<extra_id_72>": 32027,
+  "<extra_id_73>": 32026,
+  "<extra_id_74>": 32025,
+  "<extra_id_75>": 32024,
+  "<extra_id_76>": 32023,
+  "<extra_id_77>": 32022,
+  "<extra_id_78>": 32021,
+  "<extra_id_79>": 32020,
+  "<extra_id_7>": 32092,
+  "<extra_id_80>": 32019,
+  "<extra_id_81>": 32018,
+  "<extra_id_82>": 32017,
+  "<extra_id_83>": 32016,
+  "<extra_id_84>": 32015,
+  "<extra_id_85>": 32014,
+  "<extra_id_86>": 32013,
+  "<extra_id_87>": 32012,
+  "<extra_id_88>": 32011,
+  "<extra_id_89>": 32010,
+  "<extra_id_8>": 32091,
+  "<extra_id_90>": 32009,
+  "<extra_id_91>": 32008,
+  "<extra_id_92>": 32007,
+  "<extra_id_93>": 32006,
+  "<extra_id_94>": 32005,
+  "<extra_id_95>": 32004,
+  "<extra_id_96>": 32003,
+  "<extra_id_97>": 32002,
+  "<extra_id_98>": 32001,
+  "<extra_id_99>": 32000,
+  "<extra_id_9>": 32090
+}
diff --git a/tokenizer/special_tokens_map.json b/tokenizer/special_tokens_map.json
new file mode 100644
index 0000000000000000000000000000000000000000..17ade346a1042cbe0c1436f5bedcbd85c099d582
--- /dev/null
+++ b/tokenizer/special_tokens_map.json
@@ -0,0 +1,125 @@
+{
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}
diff --git a/tokenizer/spiece.model b/tokenizer/spiece.model
new file mode 100644
index 0000000000000000000000000000000000000000..317a5ccbde45300f5d1d970d4d449af2108b147e
--- /dev/null
+++ b/tokenizer/spiece.model
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d60acb128cf7b7f2536e8f38a5b18a05535c9e14c7a355904270e15b0945ea86
+size 791656
diff --git a/tokenizer/tokenizer_config.json b/tokenizer/tokenizer_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..8511c6b5feda1095f8e1115cf0dd49488b135c08
--- /dev/null
+++ b/tokenizer/tokenizer_config.json
@@ -0,0 +1,941 @@
+{
+  "add_prefix_space": true,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32000": {
+      "content": "<extra_id_99>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32001": {
+      "content": "<extra_id_98>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32002": {
+      "content": "<extra_id_97>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32003": {
+      "content": "<extra_id_96>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32004": {
+      "content": "<extra_id_95>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32005": {
+      "content": "<extra_id_94>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32006": {
+      "content": "<extra_id_93>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32007": {
+      "content": "<extra_id_92>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32008": {
+      "content": "<extra_id_91>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32009": {
+      "content": "<extra_id_90>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32010": {
+      "content": "<extra_id_89>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32011": {
+      "content": "<extra_id_88>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32012": {
+      "content": "<extra_id_87>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32013": {
+      "content": "<extra_id_86>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32014": {
+      "content": "<extra_id_85>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32015": {
+      "content": "<extra_id_84>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32016": {
+      "content": "<extra_id_83>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32017": {
+      "content": "<extra_id_82>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32018": {
+      "content": "<extra_id_81>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32019": {
+      "content": "<extra_id_80>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32020": {
+      "content": "<extra_id_79>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32021": {
+      "content": "<extra_id_78>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32022": {
+      "content": "<extra_id_77>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32023": {
+      "content": "<extra_id_76>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32024": {
+      "content": "<extra_id_75>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32025": {
+      "content": "<extra_id_74>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32026": {
+      "content": "<extra_id_73>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32027": {
+      "content": "<extra_id_72>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32028": {
+      "content": "<extra_id_71>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32029": {
+      "content": "<extra_id_70>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32030": {
+      "content": "<extra_id_69>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32031": {
+      "content": "<extra_id_68>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32032": {
+      "content": "<extra_id_67>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32033": {
+      "content": "<extra_id_66>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32034": {
+      "content": "<extra_id_65>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32035": {
+      "content": "<extra_id_64>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32036": {
+      "content": "<extra_id_63>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32037": {
+      "content": "<extra_id_62>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32038": {
+      "content": "<extra_id_61>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32039": {
+      "content": "<extra_id_60>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32040": {
+      "content": "<extra_id_59>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32041": {
+      "content": "<extra_id_58>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32042": {
+      "content": "<extra_id_57>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32043": {
+      "content": "<extra_id_56>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32044": {
+      "content": "<extra_id_55>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32045": {
+      "content": "<extra_id_54>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32046": {
+      "content": "<extra_id_53>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32047": {
+      "content": "<extra_id_52>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32048": {
+      "content": "<extra_id_51>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32049": {
+      "content": "<extra_id_50>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32050": {
+      "content": "<extra_id_49>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32051": {
+      "content": "<extra_id_48>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32052": {
+      "content": "<extra_id_47>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32053": {
+      "content": "<extra_id_46>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32054": {
+      "content": "<extra_id_45>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32055": {
+      "content": "<extra_id_44>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32056": {
+      "content": "<extra_id_43>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32057": {
+      "content": "<extra_id_42>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32058": {
+      "content": "<extra_id_41>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32059": {
+      "content": "<extra_id_40>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32060": {
+      "content": "<extra_id_39>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32061": {
+      "content": "<extra_id_38>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32062": {
+      "content": "<extra_id_37>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32063": {
+      "content": "<extra_id_36>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32064": {
+      "content": "<extra_id_35>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32065": {
+      "content": "<extra_id_34>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32066": {
+      "content": "<extra_id_33>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32067": {
+      "content": "<extra_id_32>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32068": {
+      "content": "<extra_id_31>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32069": {
+      "content": "<extra_id_30>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32070": {
+      "content": "<extra_id_29>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32071": {
+      "content": "<extra_id_28>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32072": {
+      "content": "<extra_id_27>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32073": {
+      "content": "<extra_id_26>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32074": {
+      "content": "<extra_id_25>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32075": {
+      "content": "<extra_id_24>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32076": {
+      "content": "<extra_id_23>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32077": {
+      "content": "<extra_id_22>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32078": {
+      "content": "<extra_id_21>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32079": {
+      "content": "<extra_id_20>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32080": {
+      "content": "<extra_id_19>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32081": {
+      "content": "<extra_id_18>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32082": {
+      "content": "<extra_id_17>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32083": {
+      "content": "<extra_id_16>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32084": {
+      "content": "<extra_id_15>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32085": {
+      "content": "<extra_id_14>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32086": {
+      "content": "<extra_id_13>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32087": {
+      "content": "<extra_id_12>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32088": {
+      "content": "<extra_id_11>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32089": {
+      "content": "<extra_id_10>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32090": {
+      "content": "<extra_id_9>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32091": {
+      "content": "<extra_id_8>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32092": {
+      "content": "<extra_id_7>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32093": {
+      "content": "<extra_id_6>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32094": {
+      "content": "<extra_id_5>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32095": {
+      "content": "<extra_id_4>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32096": {
+      "content": "<extra_id_3>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32097": {
+      "content": "<extra_id_2>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32098": {
+      "content": "<extra_id_1>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32099": {
+      "content": "<extra_id_0>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "extra_ids": 100,
+  "extra_special_tokens": {},
+  "legacy": true,
+  "model_max_length": 128,
+  "pad_token": "<pad>",
+  "sp_model_kwargs": {},
+  "tokenizer_class": "T5Tokenizer",
+  "unk_token": "<unk>"
+}
diff --git a/transformer/config.json b/transformer/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..8fd4f60304f8253d69adcfa6cdb943e2f8ce71a3
--- /dev/null
+++ b/transformer/config.json
@@ -0,0 +1,19 @@
+{
+  "_class_name": "LTXVideoTransformer3DModel",
+  "_diffusers_version": "0.35.0.dev0",
+  "activation_fn": "gelu-approximate",
+  "attention_bias": true,
+  "attention_head_dim": 128,
+  "attention_out_bias": true,
+  "caption_channels": 4096,
+  "cross_attention_dim": 4096,
+  "in_channels": 128,
+  "norm_elementwise_affine": false,
+  "norm_eps": 1e-06,
+  "num_attention_heads": 32,
+  "num_layers": 48,
+  "out_channels": 128,
+  "patch_size": 1,
+  "patch_size_t": 1,
+  "qk_norm": "rms_norm_across_heads"
+}
diff --git a/transformer/diffusion_pytorch_model-00001-of-00006.safetensors b/transformer/diffusion_pytorch_model-00001-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..33a7beb7db10bb77f9bc4f316738c135bde8424b
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00001-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51ca484ab70d6320374c4522c5fe6477fdea264513d8e87a4ded542fa4d6e0f8
+size 4870326248
diff --git a/transformer/diffusion_pytorch_model-00002-of-00006.safetensors b/transformer/diffusion_pytorch_model-00002-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..50c901c50b4ca494d361bb8695c68077cb0e70e9
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00002-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:57fa900382c3b2ce4375d6faea7d0514418f44339d56f63488d4244988b9d9d9
+size 4967810536
diff --git a/transformer/diffusion_pytorch_model-00003-of-00006.safetensors b/transformer/diffusion_pytorch_model-00003-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..2cf616df8657c7144edbe7caffe7198aae4285c5
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00003-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:60040ec2cf7a7f592eb6750152fdd0c4297effae79902cb304e3ae128ef0e4d8
+size 4967851880
diff --git a/transformer/diffusion_pytorch_model-00004-of-00006.safetensors b/transformer/diffusion_pytorch_model-00004-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..33fc671115fc3ef7f89ed68c6a8bfdb1d4af1e86
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00004-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:58544a1ee52213f853e43b090c87f7ff8aa3b5b465ae0e89a8e175a4518d7935
+size 4967827888
diff --git a/transformer/diffusion_pytorch_model-00005-of-00006.safetensors b/transformer/diffusion_pytorch_model-00005-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..8733e4788cbc08b063c1531eb7dee0e42f56df38
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00005-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0dcc711217c1087e62f2d5b8c72e9e45bf968863fd0c539efab8d4d57d941462
+size 4967811280
diff --git a/transformer/diffusion_pytorch_model-00006-of-00006.safetensors b/transformer/diffusion_pytorch_model-00006-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..6ccc14a0d3ef890622453774809350ca4e705c71
--- /dev/null
+++ b/transformer/diffusion_pytorch_model-00006-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db4ea47f3a60f22ab5191599d1e55cc69c619f8aa8cf54d309ac297411482a0
+size 1343650240
diff --git a/transformer/diffusion_pytorch_model.safetensors.index.json b/transformer/diffusion_pytorch_model.safetensors.index.json
new file mode 100644
index 0000000000000000000000000000000000000000..9d9d45aa1272e9900a06e3f77dcdb0ce85fda016
--- /dev/null
+++ b/transformer/diffusion_pytorch_model.safetensors.index.json
@@ -0,0 +1,1222 @@
+{
+  "metadata": {
+    "total_size": 26085138688
+  },
+  "weight_map": {
+    "caption_projection.linear_1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "caption_projection.linear_1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "caption_projection.linear_2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "caption_projection.linear_2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "proj_in.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "proj_in.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "proj_out.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.emb.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.emb.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.emb.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.emb.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.linear.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.linear.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.10.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.18.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.2.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.20.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.27.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.27.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.28.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.3.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.30.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.37.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.4.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.40.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.ff.net.0.proj.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.ff.net.2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.46.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.ff.net.0.proj.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.ff.net.2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.scale_shift_table": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.ff.net.0.proj.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.ff.net.2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.scale_shift_table": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.5.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.9.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors"
+  }
+}
diff --git a/vae/config.json b/vae/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..0d8d3069fe07761fc969a6994eff7fa6227f63d0
--- /dev/null
+++ b/vae/config.json
@@ -0,0 +1,80 @@
+{
+  "_class_name": "AutoencoderKLLTXVideo",
+  "_diffusers_version": "0.34.0.dev0",
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    1024,
+    2048
+  ],
+  "decoder_block_out_channels": [
+    256,
+    512,
+    1024
+  ],
+  "decoder_causal": false,
+  "decoder_inject_noise": [
+    false,
+    false,
+    false,
+    false
+  ],
+  "decoder_layers_per_block": [
+    5,
+    5,
+    5,
+    5
+  ],
+  "decoder_spatio_temporal_scaling": [
+    true,
+    true,
+    true
+  ],
+  "down_block_types": [
+    "LTXVideo095DownBlock3D",
+    "LTXVideo095DownBlock3D",
+    "LTXVideo095DownBlock3D",
+    "LTXVideo095DownBlock3D"
+  ],
+  "downsample_type": [
+    "spatial",
+    "temporal",
+    "spatiotemporal",
+    "spatiotemporal"
+  ],
+  "encoder_causal": true,
+  "in_channels": 3,
+  "latent_channels": 128,
+  "layers_per_block": [
+    4,
+    6,
+    6,
+    2,
+    2
+  ],
+  "out_channels": 3,
+  "patch_size": 4,
+  "patch_size_t": 1,
+  "resnet_norm_eps": 1e-06,
+  "scaling_factor": 1.0,
+  "spatial_compression_ratio": 32,
+  "spatio_temporal_scaling": [
+    true,
+    true,
+    true,
+    true
+  ],
+  "temporal_compression_ratio": 8,
+  "timestep_conditioning": true,
+  "upsample_factor": [
+    2,
+    2,
+    2
+  ],
+  "upsample_residual": [
+    true,
+    true,
+    true
+  ]
+}
diff --git a/vae/diffusion_pytorch_model.safetensors b/vae/diffusion_pytorch_model.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..5e36751cce6b6d31d5202d97907e6e2be59c133a
--- /dev/null
+++ b/vae/diffusion_pytorch_model.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3419989c7059923ee4134b858e0511d0b294a7c88b7745169763b37ebc4db7f0
+size 2493855612
diff --git a/vae/model_index.json b/vae/model_index.json
new file mode 100644
index 0000000000000000000000000000000000000000..d928724e82854203c895ee5fc956da465ffe6103
--- /dev/null
+++ b/vae/model_index.json
@@ -0,0 +1,24 @@
+{
+  "_class_name": "LTXConditionPipeline",
+  "_diffusers_version": "0.34.0.dev0",
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "T5EncoderModel"
+  ],
+  "tokenizer": [
+    "transformers",
+    "T5Tokenizer"
+  ],
+  "transformer": [
+    "diffusers",
+    "LTXVideoTransformer3DModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLLTXVideo"
+  ]
+}
diff --git a/vae/scheduler/scheduler_config.json b/vae/scheduler/scheduler_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..d100002d36531d0895d6b738044725815e5ea06c
--- /dev/null
+++ b/vae/scheduler/scheduler_config.json
@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.34.0.dev0",
+  "base_image_seq_len": 256,
+  "base_shift": 0.5,
+  "invert_sigmas": false,
+  "max_image_seq_len": 4096,
+  "max_shift": 1.15,
+  "num_train_timesteps": 1000,
+  "shift": 1.0,
+  "shift_terminal": null,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": false,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}
diff --git a/vae/text_encoder/config.json b/vae/text_encoder/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..d32c5a60ffb46d93202500fd3bea1f8a4b9ff0e7
--- /dev/null
+++ b/vae/text_encoder/config.json
@@ -0,0 +1,31 @@
+{
+  "architectures": [
+    "T5EncoderModel"
+  ],
+  "classifier_dropout": 0.0,
+  "d_ff": 10240,
+  "d_kv": 64,
+  "d_model": 4096,
+  "decoder_start_token_id": 0,
+  "dense_act_fn": "gelu_new",
+  "dropout_rate": 0.1,
+  "eos_token_id": 1,
+  "feed_forward_proj": "gated-gelu",
+  "initializer_factor": 1.0,
+  "is_encoder_decoder": true,
+  "is_gated_act": true,
+  "layer_norm_epsilon": 1e-06,
+  "model_type": "t5",
+  "num_decoder_layers": 24,
+  "num_heads": 64,
+  "num_layers": 24,
+  "output_past": true,
+  "pad_token_id": 0,
+  "relative_attention_max_distance": 128,
+  "relative_attention_num_buckets": 32,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.51.3",
+  "use_cache": true,
+  "vocab_size": 32128
+}
diff --git a/vae/text_encoder/model-00001-of-00004.safetensors b/vae/text_encoder/model-00001-of-00004.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..18c5aabb9b2b13497a1d695dca7c6aa53010ab12
--- /dev/null
+++ b/vae/text_encoder/model-00001-of-00004.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a68b2c8c080696a10109612a649bc69330991ecfea65930ccfdfbdb011f2686
+size 4989319680
diff --git a/vae/text_encoder/model-00002-of-00004.safetensors b/vae/text_encoder/model-00002-of-00004.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..99a5b982f77018015431bcac90f77fb50ef2485b
--- /dev/null
+++ b/vae/text_encoder/model-00002-of-00004.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8ed6556d7507e38af5b428c605fb2a6f2bdb7e80bd481308b865f7a40c551ca
+size 4999830656
diff --git a/vae/text_encoder/model-00003-of-00004.safetensors b/vae/text_encoder/model-00003-of-00004.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..03e07005d6976043a9bbe738c61e39a60ebabaca
--- /dev/null
+++ b/vae/text_encoder/model-00003-of-00004.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c831635f83041f83faf0024b39c6ecb21b45d70dd38a63ea5bac6c7c6e5e558c
+size 4865612720
diff --git a/vae/text_encoder/model-00004-of-00004.safetensors b/vae/text_encoder/model-00004-of-00004.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..2c6513c8a7cfd4f59f71837c19fb28bfc5aae25e
--- /dev/null
+++ b/vae/text_encoder/model-00004-of-00004.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:02a5f2d69205be92ad48fe5d712d38c2ff55627969116aeffc58bd75a28da468
+size 4194506688
diff --git a/vae/text_encoder/model.safetensors.index.json b/vae/text_encoder/model.safetensors.index.json
new file mode 100644
index 0000000000000000000000000000000000000000..0900b8ebd70052d4cccc2eacf6e07d75bb1f9cb8
--- /dev/null
+++ b/vae/text_encoder/model.safetensors.index.json
@@ -0,0 +1,226 @@
+{
+  "metadata": {
+    "total_size": 19049242624
+  },
+  "weight_map": {
+    "encoder.block.0.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.relative_attention_bias.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.0.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.1.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.10.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.11.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.12.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.12.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.12.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.13.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.14.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.15.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.16.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wi_1.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.1.DenseReluDense.wo.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.17.layer.1.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.k.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.o.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.q.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.SelfAttention.v.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.0.layer_norm.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wi_0.weight": "model-00003-of-00004.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.18.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.18.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.19.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.2.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.20.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.21.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.22.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.k.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.o.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.q.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.SelfAttention.v.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.0.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wi_0.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wi_1.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.1.DenseReluDense.wo.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.23.layer.1.layer_norm.weight": "model-00004-of-00004.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.3.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.1.DenseReluDense.wo.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.4.layer.1.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.k.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.o.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.q.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.SelfAttention.v.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.0.layer_norm.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wi_0.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wi_1.weight": "model-00001-of-00004.safetensors",
+    "encoder.block.5.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.5.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.6.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.7.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.8.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.k.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.o.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.q.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.SelfAttention.v.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.0.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wi_0.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wi_1.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.1.DenseReluDense.wo.weight": "model-00002-of-00004.safetensors",
+    "encoder.block.9.layer.1.layer_norm.weight": "model-00002-of-00004.safetensors",
+    "encoder.final_layer_norm.weight": "model-00004-of-00004.safetensors",
+    "shared.weight": "model-00001-of-00004.safetensors"
+  }
+}
diff --git a/vae/tokenizer/added_tokens.json b/vae/tokenizer/added_tokens.json
new file mode 100644
index 0000000000000000000000000000000000000000..3f5132007c4fcf42b75b65c8b6aa49c7098bcdf4
--- /dev/null
+++ b/vae/tokenizer/added_tokens.json
@@ -0,0 +1,102 @@
+{
+  "<extra_id_0>": 32099,
+  "<extra_id_10>": 32089,
+  "<extra_id_11>": 32088,
+  "<extra_id_12>": 32087,
+  "<extra_id_13>": 32086,
+  "<extra_id_14>": 32085,
+  "<extra_id_15>": 32084,
+  "<extra_id_16>": 32083,
+  "<extra_id_17>": 32082,
+  "<extra_id_18>": 32081,
+  "<extra_id_19>": 32080,
+  "<extra_id_1>": 32098,
+  "<extra_id_20>": 32079,
+  "<extra_id_21>": 32078,
+  "<extra_id_22>": 32077,
+  "<extra_id_23>": 32076,
+  "<extra_id_24>": 32075,
+  "<extra_id_25>": 32074,
+  "<extra_id_26>": 32073,
+  "<extra_id_27>": 32072,
+  "<extra_id_28>": 32071,
+  "<extra_id_29>": 32070,
+  "<extra_id_2>": 32097,
+  "<extra_id_30>": 32069,
+  "<extra_id_31>": 32068,
+  "<extra_id_32>": 32067,
+  "<extra_id_33>": 32066,
+  "<extra_id_34>": 32065,
+  "<extra_id_35>": 32064,
+  "<extra_id_36>": 32063,
+  "<extra_id_37>": 32062,
+  "<extra_id_38>": 32061,
+  "<extra_id_39>": 32060,
+  "<extra_id_3>": 32096,
+  "<extra_id_40>": 32059,
+  "<extra_id_41>": 32058,
+  "<extra_id_42>": 32057,
+  "<extra_id_43>": 32056,
+  "<extra_id_44>": 32055,
+  "<extra_id_45>": 32054,
+  "<extra_id_46>": 32053,
+  "<extra_id_47>": 32052,
+  "<extra_id_48>": 32051,
+  "<extra_id_49>": 32050,
+  "<extra_id_4>": 32095,
+  "<extra_id_50>": 32049,
+  "<extra_id_51>": 32048,
+  "<extra_id_52>": 32047,
+  "<extra_id_53>": 32046,
+  "<extra_id_54>": 32045,
+  "<extra_id_55>": 32044,
+  "<extra_id_56>": 32043,
+  "<extra_id_57>": 32042,
+  "<extra_id_58>": 32041,
+  "<extra_id_59>": 32040,
+  "<extra_id_5>": 32094,
+  "<extra_id_60>": 32039,
+  "<extra_id_61>": 32038,
+  "<extra_id_62>": 32037,
+  "<extra_id_63>": 32036,
+  "<extra_id_64>": 32035,
+  "<extra_id_65>": 32034,
+  "<extra_id_66>": 32033,
+  "<extra_id_67>": 32032,
+  "<extra_id_68>": 32031,
+  "<extra_id_69>": 32030,
+  "<extra_id_6>": 32093,
+  "<extra_id_70>": 32029,
+  "<extra_id_71>": 32028,
+  "<extra_id_72>": 32027,
+  "<extra_id_73>": 32026,
+  "<extra_id_74>": 32025,
+  "<extra_id_75>": 32024,
+  "<extra_id_76>": 32023,
+  "<extra_id_77>": 32022,
+  "<extra_id_78>": 32021,
+  "<extra_id_79>": 32020,
+  "<extra_id_7>": 32092,
+  "<extra_id_80>": 32019,
+  "<extra_id_81>": 32018,
+  "<extra_id_82>": 32017,
+  "<extra_id_83>": 32016,
+  "<extra_id_84>": 32015,
+  "<extra_id_85>": 32014,
+  "<extra_id_86>": 32013,
+  "<extra_id_87>": 32012,
+  "<extra_id_88>": 32011,
+  "<extra_id_89>": 32010,
+  "<extra_id_8>": 32091,
+  "<extra_id_90>": 32009,
+  "<extra_id_91>": 32008,
+  "<extra_id_92>": 32007,
+  "<extra_id_93>": 32006,
+  "<extra_id_94>": 32005,
+  "<extra_id_95>": 32004,
+  "<extra_id_96>": 32003,
+  "<extra_id_97>": 32002,
+  "<extra_id_98>": 32001,
+  "<extra_id_99>": 32000,
+  "<extra_id_9>": 32090
+}
diff --git a/vae/tokenizer/special_tokens_map.json b/vae/tokenizer/special_tokens_map.json
new file mode 100644
index 0000000000000000000000000000000000000000..17ade346a1042cbe0c1436f5bedcbd85c099d582
--- /dev/null
+++ b/vae/tokenizer/special_tokens_map.json
@@ -0,0 +1,125 @@
+{
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}
diff --git a/vae/tokenizer/spiece.model b/vae/tokenizer/spiece.model
new file mode 100644
index 0000000000000000000000000000000000000000..317a5ccbde45300f5d1d970d4d449af2108b147e
--- /dev/null
+++ b/vae/tokenizer/spiece.model
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d60acb128cf7b7f2536e8f38a5b18a05535c9e14c7a355904270e15b0945ea86
+size 791656
diff --git a/vae/tokenizer/tokenizer_config.json b/vae/tokenizer/tokenizer_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..8511c6b5feda1095f8e1115cf0dd49488b135c08
--- /dev/null
+++ b/vae/tokenizer/tokenizer_config.json
@@ -0,0 +1,941 @@
+{
+  "add_prefix_space": true,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32000": {
+      "content": "<extra_id_99>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32001": {
+      "content": "<extra_id_98>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32002": {
+      "content": "<extra_id_97>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32003": {
+      "content": "<extra_id_96>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32004": {
+      "content": "<extra_id_95>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32005": {
+      "content": "<extra_id_94>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32006": {
+      "content": "<extra_id_93>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32007": {
+      "content": "<extra_id_92>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32008": {
+      "content": "<extra_id_91>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32009": {
+      "content": "<extra_id_90>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32010": {
+      "content": "<extra_id_89>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32011": {
+      "content": "<extra_id_88>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32012": {
+      "content": "<extra_id_87>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32013": {
+      "content": "<extra_id_86>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32014": {
+      "content": "<extra_id_85>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32015": {
+      "content": "<extra_id_84>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32016": {
+      "content": "<extra_id_83>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32017": {
+      "content": "<extra_id_82>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32018": {
+      "content": "<extra_id_81>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32019": {
+      "content": "<extra_id_80>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32020": {
+      "content": "<extra_id_79>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32021": {
+      "content": "<extra_id_78>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32022": {
+      "content": "<extra_id_77>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32023": {
+      "content": "<extra_id_76>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32024": {
+      "content": "<extra_id_75>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32025": {
+      "content": "<extra_id_74>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32026": {
+      "content": "<extra_id_73>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32027": {
+      "content": "<extra_id_72>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32028": {
+      "content": "<extra_id_71>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32029": {
+      "content": "<extra_id_70>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32030": {
+      "content": "<extra_id_69>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32031": {
+      "content": "<extra_id_68>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32032": {
+      "content": "<extra_id_67>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32033": {
+      "content": "<extra_id_66>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32034": {
+      "content": "<extra_id_65>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32035": {
+      "content": "<extra_id_64>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32036": {
+      "content": "<extra_id_63>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32037": {
+      "content": "<extra_id_62>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32038": {
+      "content": "<extra_id_61>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32039": {
+      "content": "<extra_id_60>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32040": {
+      "content": "<extra_id_59>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32041": {
+      "content": "<extra_id_58>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32042": {
+      "content": "<extra_id_57>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32043": {
+      "content": "<extra_id_56>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32044": {
+      "content": "<extra_id_55>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32045": {
+      "content": "<extra_id_54>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32046": {
+      "content": "<extra_id_53>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32047": {
+      "content": "<extra_id_52>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32048": {
+      "content": "<extra_id_51>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32049": {
+      "content": "<extra_id_50>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32050": {
+      "content": "<extra_id_49>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32051": {
+      "content": "<extra_id_48>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32052": {
+      "content": "<extra_id_47>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32053": {
+      "content": "<extra_id_46>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32054": {
+      "content": "<extra_id_45>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32055": {
+      "content": "<extra_id_44>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32056": {
+      "content": "<extra_id_43>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32057": {
+      "content": "<extra_id_42>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32058": {
+      "content": "<extra_id_41>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32059": {
+      "content": "<extra_id_40>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32060": {
+      "content": "<extra_id_39>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32061": {
+      "content": "<extra_id_38>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32062": {
+      "content": "<extra_id_37>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32063": {
+      "content": "<extra_id_36>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32064": {
+      "content": "<extra_id_35>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32065": {
+      "content": "<extra_id_34>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32066": {
+      "content": "<extra_id_33>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32067": {
+      "content": "<extra_id_32>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32068": {
+      "content": "<extra_id_31>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32069": {
+      "content": "<extra_id_30>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32070": {
+      "content": "<extra_id_29>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32071": {
+      "content": "<extra_id_28>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32072": {
+      "content": "<extra_id_27>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32073": {
+      "content": "<extra_id_26>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32074": {
+      "content": "<extra_id_25>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32075": {
+      "content": "<extra_id_24>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32076": {
+      "content": "<extra_id_23>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32077": {
+      "content": "<extra_id_22>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32078": {
+      "content": "<extra_id_21>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32079": {
+      "content": "<extra_id_20>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32080": {
+      "content": "<extra_id_19>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32081": {
+      "content": "<extra_id_18>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32082": {
+      "content": "<extra_id_17>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32083": {
+      "content": "<extra_id_16>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32084": {
+      "content": "<extra_id_15>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32085": {
+      "content": "<extra_id_14>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32086": {
+      "content": "<extra_id_13>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32087": {
+      "content": "<extra_id_12>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32088": {
+      "content": "<extra_id_11>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32089": {
+      "content": "<extra_id_10>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32090": {
+      "content": "<extra_id_9>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32091": {
+      "content": "<extra_id_8>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32092": {
+      "content": "<extra_id_7>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32093": {
+      "content": "<extra_id_6>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32094": {
+      "content": "<extra_id_5>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32095": {
+      "content": "<extra_id_4>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32096": {
+      "content": "<extra_id_3>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32097": {
+      "content": "<extra_id_2>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32098": {
+      "content": "<extra_id_1>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    },
+    "32099": {
+      "content": "<extra_id_0>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": true,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "extra_ids": 100,
+  "extra_special_tokens": {},
+  "legacy": true,
+  "model_max_length": 128,
+  "pad_token": "<pad>",
+  "sp_model_kwargs": {},
+  "tokenizer_class": "T5Tokenizer",
+  "unk_token": "<unk>"
+}
diff --git a/vae/transformer/config.json b/vae/transformer/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..8fd4f60304f8253d69adcfa6cdb943e2f8ce71a3
--- /dev/null
+++ b/vae/transformer/config.json
@@ -0,0 +1,19 @@
+{
+  "_class_name": "LTXVideoTransformer3DModel",
+  "_diffusers_version": "0.35.0.dev0",
+  "activation_fn": "gelu-approximate",
+  "attention_bias": true,
+  "attention_head_dim": 128,
+  "attention_out_bias": true,
+  "caption_channels": 4096,
+  "cross_attention_dim": 4096,
+  "in_channels": 128,
+  "norm_elementwise_affine": false,
+  "norm_eps": 1e-06,
+  "num_attention_heads": 32,
+  "num_layers": 48,
+  "out_channels": 128,
+  "patch_size": 1,
+  "patch_size_t": 1,
+  "qk_norm": "rms_norm_across_heads"
+}
diff --git a/vae/transformer/diffusion_pytorch_model-00001-of-00006.safetensors b/vae/transformer/diffusion_pytorch_model-00001-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..33a7beb7db10bb77f9bc4f316738c135bde8424b
--- /dev/null
+++ b/vae/transformer/diffusion_pytorch_model-00001-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51ca484ab70d6320374c4522c5fe6477fdea264513d8e87a4ded542fa4d6e0f8
+size 4870326248
diff --git a/vae/transformer/diffusion_pytorch_model-00002-of-00006.safetensors b/vae/transformer/diffusion_pytorch_model-00002-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..50c901c50b4ca494d361bb8695c68077cb0e70e9
--- /dev/null
+++ b/vae/transformer/diffusion_pytorch_model-00002-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:57fa900382c3b2ce4375d6faea7d0514418f44339d56f63488d4244988b9d9d9
+size 4967810536
diff --git a/vae/transformer/diffusion_pytorch_model-00003-of-00006.safetensors b/vae/transformer/diffusion_pytorch_model-00003-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..2cf616df8657c7144edbe7caffe7198aae4285c5
--- /dev/null
+++ b/vae/transformer/diffusion_pytorch_model-00003-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:60040ec2cf7a7f592eb6750152fdd0c4297effae79902cb304e3ae128ef0e4d8
+size 4967851880
diff --git a/vae/transformer/diffusion_pytorch_model-00004-of-00006.safetensors b/vae/transformer/diffusion_pytorch_model-00004-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..33fc671115fc3ef7f89ed68c6a8bfdb1d4af1e86
--- /dev/null
+++ b/vae/transformer/diffusion_pytorch_model-00004-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:58544a1ee52213f853e43b090c87f7ff8aa3b5b465ae0e89a8e175a4518d7935
+size 4967827888
diff --git a/vae/transformer/diffusion_pytorch_model-00005-of-00006.safetensors b/vae/transformer/diffusion_pytorch_model-00005-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..8733e4788cbc08b063c1531eb7dee0e42f56df38
--- /dev/null
+++ b/vae/transformer/diffusion_pytorch_model-00005-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0dcc711217c1087e62f2d5b8c72e9e45bf968863fd0c539efab8d4d57d941462
+size 4967811280
diff --git a/vae/transformer/diffusion_pytorch_model-00006-of-00006.safetensors b/vae/transformer/diffusion_pytorch_model-00006-of-00006.safetensors
new file mode 100644
index 0000000000000000000000000000000000000000..6ccc14a0d3ef890622453774809350ca4e705c71
--- /dev/null
+++ b/vae/transformer/diffusion_pytorch_model-00006-of-00006.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db4ea47f3a60f22ab5191599d1e55cc69c619f8aa8cf54d309ac297411482a0
+size 1343650240
diff --git a/vae/transformer/diffusion_pytorch_model.safetensors.index.json b/vae/transformer/diffusion_pytorch_model.safetensors.index.json
new file mode 100644
index 0000000000000000000000000000000000000000..9d9d45aa1272e9900a06e3f77dcdb0ce85fda016
--- /dev/null
+++ b/vae/transformer/diffusion_pytorch_model.safetensors.index.json
@@ -0,0 +1,1222 @@
+{
+  "metadata": {
+    "total_size": 26085138688
+  },
+  "weight_map": {
+    "caption_projection.linear_1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "caption_projection.linear_1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "caption_projection.linear_2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "caption_projection.linear_2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "proj_in.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "proj_in.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "proj_out.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "proj_out.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.emb.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.emb.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.emb.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.emb.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.linear.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "time_embed.linear.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.10.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.18.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.2.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.20.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.25.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn1.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_k.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_out.0.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_out.0.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_q.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_v.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.attn2.to_v.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.26.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.27.attn1.norm_k.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.27.attn1.norm_q.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.28.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.3.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.30.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.attn2.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.35.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_k.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_out.0.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_out.0.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_q.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_v.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn1.to_v.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn2.norm_k.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn2.norm_q.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.37.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.4.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.40.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.42.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.43.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.44.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn1.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.norm_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.norm_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_k.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_k.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_out.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_out.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_q.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_q.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_v.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.attn2.to_v.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.45.ff.net.0.proj.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.ff.net.2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.46.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.ff.net.0.proj.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.ff.net.2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.scale_shift_table": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.norm_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.norm_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_k.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_out.0.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_out.0.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_q.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_v.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.to_v.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.norm_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.norm_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_k.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_k.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_out.0.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_out.0.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_q.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_q.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_v.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.to_v.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.ff.net.0.proj.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.ff.net.2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.scale_shift_table": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.5.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.ff.net.0.proj.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.ff.net.2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.ff.net.2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.norm_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.norm_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_k.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_k.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_out.0.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_out.0.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_q.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_q.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_v.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.to_v.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.9.attn1.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.norm_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.norm_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_k.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_k.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_out.0.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_out.0.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_q.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_q.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_v.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.to_v.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.ff.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.ff.net.2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.ff.net.2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors"
+  }
+}