Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,10 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+lib/liboptimus_ths-torch2.2-cu121.cpython-310-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
+lib/liboptimus_ths-torch2.3-cu121.cpython-310-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
+lib/liboptimus_ths-torch2.5-cu124.cpython-310-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
+assets/dcvae.png filter=lfs diff=lfs merge=lfs -text
+assets/dit.png filter=lfs diff=lfs merge=lfs -text
+assets/dpo_pipeline.png filter=lfs diff=lfs merge=lfs -text
+assets/model_architecture.png filter=lfs diff=lfs merge=lfs -text

hunyuan_clip/clip_text_encoder/config.json

@@ -0,0 +1,34 @@
+{
+  "_name_or_path": "hfl/chinese-roberta-wwm-ext-large",
+  "architectures": [
+    "BertModel"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "bos_token_id": 0,
+  "classifier_dropout": null,
+  "directionality": "bidi",
+  "eos_token_id": 2,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 1024,
+  "initializer_range": 0.02,
+  "intermediate_size": 4096,
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "bert",
+  "num_attention_heads": 16,
+  "num_hidden_layers": 24,
+  "output_past": true,
+  "pad_token_id": 0,
+  "pooler_fc_size": 768,
+  "pooler_num_attention_heads": 12,
+  "pooler_num_fc_layers": 3,
+  "pooler_size_per_head": 128,
+  "pooler_type": "first_token_transform",
+  "position_embedding_type": "absolute",
+  "torch_dtype": "float32",
+  "transformers_version": "4.22.1",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 47020
+}

hunyuan_clip/clip_text_encoder/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77fd65c751513310fe575e0ef59963743f89a4955ee5c79d7468157b27e83c51
+size 3936679395

hunyuan_clip/tokenizer/special_tokens_map.json

@@ -0,0 +1,7 @@
+{
+  "cls_token": "[CLS]",
+  "mask_token": "[MASK]",
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "unk_token": "[UNK]"
+}

hunyuan_clip/tokenizer/tokenizer_config.json

@@ -0,0 +1,16 @@
+{
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": true,
+  "mask_token": "[MASK]",
+  "name_or_path": "hfl/chinese-roberta-wwm-ext",
+  "never_split": null,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "special_tokens_map_file": "/home/chenweifeng/.cache/huggingface/hub/models--hfl--chinese-roberta-wwm-ext/snapshots/5c58d0b8ec1d9014354d691c538661bf00bfdb44/special_tokens_map.json",
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "unk_token": "[UNK]",
+  "model_max_length": 77
+}

lib/liboptimus_ths-torch2.2-cu121.cpython-310-x86_64-linux-gnu.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2201371a93b103f7a5bc87dc8b9f1d781a81fc272d5a8ac8c1a82877aea98661
+size 3994424

lib/liboptimus_ths-torch2.3-cu121.cpython-310-x86_64-linux-gnu.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a43a22318a90192fe37e167689bbfda9321b7eeb004f8cbafe671fe99204f1fe
+size 3998552

lib/liboptimus_ths-torch2.5-cu124.cpython-310-x86_64-linux-gnu.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab934f8ca508ec04b252674d3654628b1eea39ce7834f004a71c6fae399b9ce6
+size 4002768

model_index.json

@@ -0,0 +1,16 @@
+{
+  "_class_name": "StepVideoPipeline",
+  "_diffusers_version": "0.31.0",
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "transformer": [
+    "diffusers",
+    "StepVideoModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLStepvideo"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,8 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.31.0",
+  "device": null,
+  "num_train_timesteps": 1000,
+  "reverse": false,
+  "solver": "euler"
+}

step_llm/config.json

@@ -0,0 +1,32 @@
+{
+  "_name_or_path": "/mnt/shared-storage/tenant/opensource/step_llm",
+  "allow_transformer_engine": false,
+  "architectures": [
+    "Step1Model"
+  ],
+  "attention_dropout": 0.0,
+  "attention_impl": "GQA",
+  "base_batch_size": 128,
+  "embedding_weights_in_fp32": false,
+  "ffn_hidden_size": 16896,
+  "fp32_residual_connection": false,
+  "hidden_dropout": 0.0,
+  "hidden_size": 6144,
+  "kv_channels": 128,
+  "layernorm_epsilon": 1e-05,
+  "max_position_embeddings": 16384,
+  "num_attention_groups": 8,
+  "num_attention_heads": 48,
+  "num_layers": 48,
+  "orig_vocab_size": 65536,
+  "overlap_p2p_comm": true,
+  "padded_vocab_size": 65536,
+  "params_dtype": "torch.bfloat16",
+  "seq_length": 16384,
+  "swiglu_recompute_silu_dot": true,
+  "tokens_to_generate": 512,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.48.3",
+  "use_flash_attn": true,
+  "virtual_pipeline_model_parallel_size": 3
+}

step_llm/model-00001-of-00009.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0dc79bad96b5f9fdd56106864c56fdc7955627062de4d023ee99944ddac0c930
+size 4976668520

step_llm/model-00003-of-00009.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a7bd774bd5c30b51ae342ae92ae9d2706d631bf4ee817fd0997ccc1b0b651c3
+size 4794241248

step_llm/model-00004-of-00009.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4c025fd3756fad22460e4c3796fbb89d29ee4819c00bd58934ae8853c1319e3d
+size 3258974208

step_llm/model-00005-of-00009.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bfe129667571adf93917a54839a7af52f702bd7c7e63aee34912228507d6c1dc
+size 4794241248

step_llm/model-00006-of-00009.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b4f01bd89d7e047de600579071e6e5e17dfff7aaf8143323ab2989ec840db26
+size 4794241248

step_llm/model-00008-of-00009.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:40c5fd462b5422bdb8665fcb2e4c461cbdbd39f92c9b5e677f87a77044c259af
+size 4794241248

step_llm/model-00009-of-00009.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d707aa08cd795a137e6b18558658e65bd9969a27a8aca34d201e3a1b1a68fa4e
+size 622879216

step_llm/model.safetensors.index.json

@@ -0,0 +1,296 @@
+{
+  "metadata": {
+    "total_size": 39159201792
+  },
+  "weight_map": {
+    "tok_embeddings.word_embeddings.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.0.attention.wo.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.0.attention.wqkv.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.0.attention_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.0.feed_forward.w1.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.0.feed_forward.w2.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.0.ffn_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.1.attention.wo.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.1.attention.wqkv.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.1.attention_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.1.feed_forward.w1.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.1.feed_forward.w2.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.1.ffn_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.10.attention.wo.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.10.attention.wqkv.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.10.attention_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.10.feed_forward.w1.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.10.feed_forward.w2.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.10.ffn_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.11.attention.wo.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.11.attention.wqkv.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.11.attention_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.11.feed_forward.w1.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.11.feed_forward.w2.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.11.ffn_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.12.attention.wo.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.12.attention.wqkv.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.12.attention_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.12.feed_forward.w1.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.12.feed_forward.w2.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.12.ffn_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.13.attention.wo.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.13.attention.wqkv.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.13.attention_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.13.feed_forward.w1.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.13.feed_forward.w2.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.13.ffn_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.14.attention.wo.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.14.attention.wqkv.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.14.attention_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.14.feed_forward.w1.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.14.feed_forward.w2.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.14.ffn_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.15.attention.wo.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.15.attention.wqkv.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.15.attention_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.15.feed_forward.w1.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.15.feed_forward.w2.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.15.ffn_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.16.attention.wo.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.16.attention.wqkv.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.16.attention_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.16.feed_forward.w1.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.16.feed_forward.w2.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.16.ffn_norm.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.17.attention.wo.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.17.attention.wqkv.weight": "model-00003-of-00009.safetensors",
+    "transformer.layers.17.attention_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.17.feed_forward.w1.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.17.feed_forward.w2.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.17.ffn_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.18.attention.wo.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.18.attention.wqkv.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.18.attention_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.18.feed_forward.w1.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.18.feed_forward.w2.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.18.ffn_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.19.attention.wo.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.19.attention.wqkv.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.19.attention_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.19.feed_forward.w1.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.19.feed_forward.w2.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.19.ffn_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.2.attention.wo.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.2.attention.wqkv.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.2.attention_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.2.feed_forward.w1.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.2.feed_forward.w2.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.2.ffn_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.20.attention.wo.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.20.attention.wqkv.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.20.attention_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.20.feed_forward.w1.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.20.feed_forward.w2.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.20.ffn_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.21.attention.wo.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.21.attention.wqkv.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.21.attention_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.21.feed_forward.w1.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.21.feed_forward.w2.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.21.ffn_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.22.attention.wo.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.22.attention.wqkv.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.22.attention_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.22.feed_forward.w1.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.22.feed_forward.w2.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.22.ffn_norm.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.23.attention.wo.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.23.attention.wqkv.weight": "model-00004-of-00009.safetensors",
+    "transformer.layers.23.attention_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.23.feed_forward.w1.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.23.feed_forward.w2.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.23.ffn_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.24.attention.wo.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.24.attention.wqkv.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.24.attention_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.24.feed_forward.w1.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.24.feed_forward.w2.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.24.ffn_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.25.attention.wo.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.25.attention.wqkv.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.25.attention_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.25.feed_forward.w1.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.25.feed_forward.w2.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.25.ffn_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.26.attention.wo.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.26.attention.wqkv.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.26.attention_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.26.feed_forward.w1.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.26.feed_forward.w2.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.26.ffn_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.27.attention.wo.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.27.attention.wqkv.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.27.attention_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.27.feed_forward.w1.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.27.feed_forward.w2.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.27.ffn_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.28.attention.wo.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.28.attention.wqkv.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.28.attention_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.28.feed_forward.w1.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.28.feed_forward.w2.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.28.ffn_norm.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.29.attention.wo.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.29.attention.wqkv.weight": "model-00005-of-00009.safetensors",
+    "transformer.layers.29.attention_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.29.feed_forward.w1.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.29.feed_forward.w2.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.29.ffn_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.3.attention.wo.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.3.attention.wqkv.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.3.attention_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.3.feed_forward.w1.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.3.feed_forward.w2.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.3.ffn_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.30.attention.wo.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.30.attention.wqkv.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.30.attention_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.30.feed_forward.w1.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.30.feed_forward.w2.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.30.ffn_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.31.attention.wo.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.31.attention.wqkv.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.31.attention_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.31.feed_forward.w1.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.31.feed_forward.w2.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.31.ffn_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.32.attention.wo.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.32.attention.wqkv.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.32.attention_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.32.feed_forward.w1.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.32.feed_forward.w2.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.32.ffn_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.33.attention.wo.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.33.attention.wqkv.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.33.attention_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.33.feed_forward.w1.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.33.feed_forward.w2.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.33.ffn_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.34.attention.wo.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.34.attention.wqkv.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.34.attention_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.34.feed_forward.w1.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.34.feed_forward.w2.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.34.ffn_norm.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.35.attention.wo.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.35.attention.wqkv.weight": "model-00006-of-00009.safetensors",
+    "transformer.layers.35.attention_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.35.feed_forward.w1.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.35.feed_forward.w2.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.35.ffn_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.36.attention.wo.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.36.attention.wqkv.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.36.attention_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.36.feed_forward.w1.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.36.feed_forward.w2.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.36.ffn_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.37.attention.wo.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.37.attention.wqkv.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.37.attention_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.37.feed_forward.w1.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.37.feed_forward.w2.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.37.ffn_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.38.attention.wo.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.38.attention.wqkv.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.38.attention_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.38.feed_forward.w1.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.38.feed_forward.w2.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.38.ffn_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.39.attention.wo.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.39.attention.wqkv.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.39.attention_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.39.feed_forward.w1.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.39.feed_forward.w2.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.39.ffn_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.4.attention.wo.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.4.attention.wqkv.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.4.attention_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.4.feed_forward.w1.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.4.feed_forward.w2.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.4.ffn_norm.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.40.attention.wo.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.40.attention.wqkv.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.40.attention_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.40.feed_forward.w1.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.40.feed_forward.w2.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.40.ffn_norm.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.41.attention.wo.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.41.attention.wqkv.weight": "model-00007-of-00009.safetensors",
+    "transformer.layers.41.attention_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.41.feed_forward.w1.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.41.feed_forward.w2.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.41.ffn_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.42.attention.wo.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.42.attention.wqkv.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.42.attention_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.42.feed_forward.w1.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.42.feed_forward.w2.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.42.ffn_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.43.attention.wo.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.43.attention.wqkv.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.43.attention_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.43.feed_forward.w1.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.43.feed_forward.w2.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.43.ffn_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.44.attention.wo.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.44.attention.wqkv.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.44.attention_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.44.feed_forward.w1.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.44.feed_forward.w2.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.44.ffn_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.45.attention.wo.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.45.attention.wqkv.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.45.attention_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.45.feed_forward.w1.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.45.feed_forward.w2.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.45.ffn_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.46.attention.wo.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.46.attention.wqkv.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.46.attention_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.46.feed_forward.w1.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.46.feed_forward.w2.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.46.ffn_norm.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.47.attention.wo.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.47.attention.wqkv.weight": "model-00008-of-00009.safetensors",
+    "transformer.layers.47.attention_norm.weight": "model-00009-of-00009.safetensors",
+    "transformer.layers.47.feed_forward.w1.weight": "model-00009-of-00009.safetensors",
+    "transformer.layers.47.feed_forward.w2.weight": "model-00009-of-00009.safetensors",
+    "transformer.layers.47.ffn_norm.weight": "model-00009-of-00009.safetensors",
+    "transformer.layers.5.attention.wo.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.5.attention.wqkv.weight": "model-00001-of-00009.safetensors",
+    "transformer.layers.5.attention_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.5.feed_forward.w1.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.5.feed_forward.w2.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.5.ffn_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.6.attention.wo.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.6.attention.wqkv.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.6.attention_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.6.feed_forward.w1.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.6.feed_forward.w2.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.6.ffn_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.7.attention.wo.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.7.attention.wqkv.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.7.attention_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.7.feed_forward.w1.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.7.feed_forward.w2.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.7.ffn_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.8.attention.wo.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.8.attention.wqkv.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.8.attention_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.8.feed_forward.w1.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.8.feed_forward.w2.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.8.ffn_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.9.attention.wo.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.9.attention.wqkv.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.9.attention_norm.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.9.feed_forward.w1.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.9.feed_forward.w2.weight": "model-00002-of-00009.safetensors",
+    "transformer.layers.9.ffn_norm.weight": "model-00002-of-00009.safetensors"
+  }
+}

step_llm/step1_chat_tokenizer.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae238f74162b98d3384eb1ceef143ed67f822528a6b0d6c5e351f87d7f4a651a
+size 1053360

transformer/config.json

@@ -0,0 +1,20 @@
+{
+  "_class_name": "StepVideoModel",
+  "_diffusers_version": "0.31.0",
+  "attention_head_dim": 128,
+  "attention_type": "parallel",
+  "caption_channels": [
+    6144,
+    1024
+  ],
+  "dropout": 0.0,
+  "in_channels": 64,
+  "norm_elementwise_affine": false,
+  "norm_eps": 1e-06,
+  "norm_type": "ada_norm_single",
+  "num_attention_heads": 48,
+  "num_layers": 48,
+  "out_channels": 64,
+  "patch_size": 1,
+  "use_additional_conditions": false
+}

transformer/diffusion_pytorch_model-00001-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:09c5706674cfb0738b9289289430338e3dc53ef650d7987149c37849985a33c7
+size 9967619936

transformer/diffusion_pytorch_model-00002-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d97e0f83459c2dad63fb9c928ab084e8001c0e6a77025d67efa33bd928f32055
+size 9966673176

transformer/diffusion_pytorch_model-00003-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8353cfe94f18e31b9e7b4b36d2b6fcdf31a69d29839d57c83af25143bfb8ba29
+size 9966672264

transformer/diffusion_pytorch_model-00004-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ef9aafe3e436b2469a27e04fb1539f342bf47e783308b537e555aec680059e86
+size 9966770888

transformer/diffusion_pytorch_model-00005-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c05dcc816ca5d62991581409d5e4cbc5922fda8734e96acb4c6609a911705edf
+size 9966697872

transformer/diffusion_pytorch_model-00006-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b06c7ff620dfcaf5f757f497edb50cbb9e746f35b5645c9c161c8051cec44a81
+size 8850611776

transformer/diffusion_pytorch_model.safetensors.index.json

@@ -0,0 +1,792 @@
+{
+  "metadata": {
+    "total_size": 58684956800
+  },
+  "weight_map": {
+    "adaln_single.emb.timestep_embedder.linear_1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "adaln_single.emb.timestep_embedder.linear_1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "adaln_single.emb.timestep_embedder.linear_2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "adaln_single.emb.timestep_embedder.linear_2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "adaln_single.linear.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "adaln_single.linear.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "caption_projection.linear_1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "caption_projection.linear_1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "caption_projection.linear_2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "caption_projection.linear_2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "clip_projection.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "clip_projection.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "pos_embed.proj.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "pos_embed.proj.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",
+    "transformer_blocks.0.attn1.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.wkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.0.norm2.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.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.wkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.1.norm2.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.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn1.wqkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.norm1.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.norm1.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.norm2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.10.norm2.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.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn1.wqkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.norm1.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.norm1.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.norm2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.11.norm2.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.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn1.wqkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.norm1.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.norm1.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.norm2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.12.norm2.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.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn1.wqkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.norm1.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.norm1.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.norm2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.13.norm2.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.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn1.wqkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.norm1.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.norm1.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.norm2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.14.norm2.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.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn1.wqkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.norm1.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.norm1.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.norm2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.15.norm2.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.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn1.wqkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.attn2.wq.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.16.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.16.norm1.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.norm1.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.norm2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.16.norm2.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.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.attn1.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.attn1.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.attn1.wqkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.attn2.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.attn2.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.attn2.wkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.attn2.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.attn2.wq.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.ff.net.2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.norm1.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.norm1.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.norm2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.norm2.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.17.scale_shift_table": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn1.wqkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.wkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.norm1.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.norm1.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.norm2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.18.norm2.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.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn1.wqkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.wkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.norm1.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.norm1.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.norm2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.19.norm2.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.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.wkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.2.norm2.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.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn1.wqkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.wkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.norm1.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.norm1.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.norm2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.20.norm2.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.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn1.wqkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.wkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.norm1.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.norm1.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.norm2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.21.norm2.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.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn1.wqkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.wkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.norm1.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.norm1.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.norm2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.22.norm2.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.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn1.wqkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.wkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.norm1.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.norm1.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.norm2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.23.norm2.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.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn1.wqkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.k_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.q_norm.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.wkv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.24.norm1.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.norm1.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.norm2.bias": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "transformer_blocks.24.norm2.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.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.attn1.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.attn1.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.attn1.wqkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.attn2.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.attn2.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.attn2.wkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.attn2.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.attn2.wq.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.norm2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.norm2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.25.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn1.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn1.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn1.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn1.wqkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn2.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn2.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn2.wkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn2.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.attn2.wq.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.0.proj.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.ff.net.2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.norm2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.norm2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.26.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn1.wqkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.wkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.norm2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.norm2.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.27.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn1.wqkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.wkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.norm2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.28.norm2.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.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn1.wqkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.wkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.norm2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.29.norm2.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.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.wkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.3.norm2.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.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn1.wqkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.wkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.norm2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.30.norm2.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.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn1.wqkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.wkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.norm2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.31.norm2.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.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn1.wqkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.k_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.q_norm.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.wkv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.norm2.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.32.norm2.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.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.attn2.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.attn2.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.attn2.wkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.attn2.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.attn2.wq.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.norm1.bias": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.norm1.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.33.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.norm2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.33.scale_shift_table": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "transformer_blocks.34.attn1.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.attn2.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.attn2.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.attn2.wkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.attn2.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.attn2.wq.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.norm1.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.norm1.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.norm2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.34.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn1.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn2.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn2.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn2.wkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn2.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.attn2.wq.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.0.proj.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.ff.net.2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.norm1.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.norm1.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.norm2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.35.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn1.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.wkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.norm1.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.norm1.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.norm2.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.36.scale_shift_table": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.wkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.norm1.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.norm1.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.37.norm2.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.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.wkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.norm1.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.norm1.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.38.norm2.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.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.wkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.norm1.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.norm1.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.39.norm2.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.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.wkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.4.norm2.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.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.wkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.norm1.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.norm1.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.40.norm2.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.k_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.q_norm.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn1.wqkv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.attn2.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.41.attn2.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.41.attn2.wkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.41.attn2.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.41.attn2.wq.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.41.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.41.norm1.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.norm1.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.norm2.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "transformer_blocks.41.norm2.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.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.attn1.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.attn1.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.attn1.wqkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.attn2.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.attn2.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.attn2.wkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.attn2.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.attn2.wq.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.norm1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.norm1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.norm2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.norm2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.42.scale_shift_table": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn1.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn1.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn1.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn1.wqkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn2.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn2.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn2.wkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn2.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.attn2.wq.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.norm1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.norm1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.norm2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.norm2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.43.scale_shift_table": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn1.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn1.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn1.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn1.wqkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn2.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn2.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn2.wkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn2.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.attn2.wq.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.0.proj.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.ff.net.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.norm1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.norm1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.norm2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.norm2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.44.scale_shift_table": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn1.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn1.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn1.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn1.wqkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn2.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn2.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn2.wkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn2.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.norm1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.norm1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.norm2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.norm2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.45.scale_shift_table": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn1.wqkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.wkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.norm1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.norm1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.norm2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.46.norm2.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.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn1.wqkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.k_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.q_norm.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.wkv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.norm1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.norm1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.norm2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "transformer_blocks.47.norm2.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.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.wkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.5.norm2.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.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.wkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.6.norm2.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.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.wkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.7.norm2.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.k_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.q_norm.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn1.wqkv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.8.norm1.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.norm1.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.norm2.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.norm2.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.8.scale_shift_table": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "transformer_blocks.9.attn1.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn1.wqkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.k_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.q_norm.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.wkv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.attn2.wq.weight": "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.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.norm1.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.norm1.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.norm2.bias": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.norm2.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "transformer_blocks.9.scale_shift_table": "diffusion_pytorch_model-00002-of-00006.safetensors"
+  }
+}

vae/config.json

@@ -0,0 +1,14 @@
+{
+    "_class_name": "AutoencoderKLStepvideo",
+    "_diffusers_version": "0.31.0",
+    "in_channels": 3,
+    "out_channels": 3,
+    "z_channels": 64,
+    "num_res_blocks": 2,
+    "version": 2,
+    "frame_len": 17,
+    "spatial_compression_ratio": 16,
+    "temporal_compression_ratio": 4,
+    "scaling_factor": 1.0
+  }
+  

vae/v1_vae/vae.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:abb2747d01c482844712e2660a723c76d8f9f7031d85411a4eb74a5566daa88f
+size 1035340564

vae/vae_v2.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c8fa89118d7f615c4383e5820e060355ce8ebcd8865eca3f7797cb808b974cd
+size 1995304052