Add files using upload-large-folder tool

.gitattributes

@@ -33,3 +33,4 @@ 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
+tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text

model_index.json

@@ -0,0 +1,24 @@
+{
+  "_class_name": "Krea2Pipeline",
+  "_diffusers_version": "0.39.0.dev0",
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "Qwen3VLModel"
+  ],
+  "tokenizer": [
+    "transformers",
+    "Qwen2Tokenizer"
+  ],
+  "transformer": [
+    "diffusers",
+    "Krea2Transformer2DModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLQwenImage"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.39.0.dev0",
+  "base_image_seq_len": 256,
+  "base_shift": 1.15,
+  "invert_sigmas": false,
+  "max_image_seq_len": 6400,
+  "max_shift": 1.15,
+  "num_train_timesteps": 1000,
+  "shift": 1.0,
+  "shift_terminal": null,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": true,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}

text_encoder/config.json

@@ -0,0 +1,66 @@
+{
+  "architectures": [
+    "Qwen3VLModel"
+  ],
+  "dtype": "bfloat16",
+  "image_token_id": 151655,
+  "model_type": "qwen3_vl",
+  "text_config": {
+    "attention_bias": false,
+    "attention_dropout": 0.0,
+    "bos_token_id": 151643,
+    "dtype": "bfloat16",
+    "eos_token_id": 151645,
+    "head_dim": 128,
+    "hidden_act": "silu",
+    "hidden_size": 2560,
+    "initializer_range": 0.02,
+    "intermediate_size": 9728,
+    "max_position_embeddings": 262144,
+    "model_type": "qwen3_vl_text",
+    "num_attention_heads": 32,
+    "num_hidden_layers": 36,
+    "num_key_value_heads": 8,
+    "pad_token_id": null,
+    "rms_norm_eps": 1e-06,
+    "rope_parameters": {
+      "mrope_interleaved": true,
+      "mrope_section": [
+        24,
+        20,
+        20
+      ],
+      "rope_theta": 5000000,
+      "rope_type": "default"
+    },
+    "tie_word_embeddings": true,
+    "use_cache": true,
+    "vocab_size": 151936
+  },
+  "tie_word_embeddings": true,
+  "transformers_version": "5.13.0.dev0",
+  "video_token_id": 151656,
+  "vision_config": {
+    "deepstack_visual_indexes": [
+      5,
+      11,
+      17
+    ],
+    "depth": 24,
+    "dtype": "bfloat16",
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 1024,
+    "in_channels": 3,
+    "initializer_range": 0.02,
+    "intermediate_size": 4096,
+    "model_type": "qwen3_vl_vision",
+    "num_heads": 16,
+    "num_position_embeddings": 2304,
+    "out_hidden_size": 2560,
+    "patch_size": 16,
+    "spatial_merge_size": 2,
+    "temporal_patch_size": 2
+  },
+  "vision_end_token_id": 151653,
+  "vision_start_token_id": 151652
+}

text_encoder/model.safetensors

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

tokenizer/chat_template.jinja

@@ -0,0 +1,120 @@
+{%- if tools %}
+    {{- '<|im_start|>system\n' }}
+    {%- if messages[0].role == 'system' %}
+        {%- if messages[0].content is string %}
+            {{- messages[0].content }}
+        {%- else %}
+            {%- for content in messages[0].content %}
+                {%- if 'text' in content %}
+                    {{- content.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {{- '\n\n' }}
+    {%- endif %}
+    {{- "# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
+    {%- for tool in tools %}
+        {{- "\n" }}
+        {{- tool | tojson }}
+    {%- endfor %}
+    {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
+{%- else %}
+    {%- if messages[0].role == 'system' %}
+        {{- '<|im_start|>system\n' }}
+        {%- if messages[0].content is string %}
+            {{- messages[0].content }}
+        {%- else %}
+            {%- for content in messages[0].content %}
+                {%- if 'text' in content %}
+                    {{- content.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {{- '<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- set image_count = namespace(value=0) %}
+{%- set video_count = namespace(value=0) %}
+{%- for message in messages %}
+    {%- if message.role == "user" %}
+        {{- '<|im_start|>' + message.role + '\n' }}
+        {%- if message.content is string %}
+            {{- message.content }}
+        {%- else %}
+            {%- for content in message.content %}
+                {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
+                    {%- set image_count.value = image_count.value + 1 %}
+                    {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
+                    <|vision_start|><|image_pad|><|vision_end|>
+                {%- elif content.type == 'video' or 'video' in content %}
+                    {%- set video_count.value = video_count.value + 1 %}
+                    {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
+                    <|vision_start|><|video_pad|><|vision_end|>
+                {%- elif 'text' in content %}
+                    {{- content.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "assistant" %}
+        {{- '<|im_start|>' + message.role + '\n' }}
+        {%- if message.content is string %}
+            {{- message.content }}
+        {%- else %}
+            {%- for content_item in message.content %}
+                {%- if 'text' in content_item %}
+                    {{- content_item.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {%- if message.tool_calls %}
+            {%- for tool_call in message.tool_calls %}
+                {%- if (loop.first and message.content) or (not loop.first) %}
+                    {{- '\n' }}
+                {%- endif %}
+                {%- if tool_call.function %}
+                    {%- set tool_call = tool_call.function %}
+                {%- endif %}
+                {{- '<tool_call>\n{"name": "' }}
+                {{- tool_call.name }}
+                {{- '", "arguments": ' }}
+                {%- if tool_call.arguments is string %}
+                    {{- tool_call.arguments }}
+                {%- else %}
+                    {{- tool_call.arguments | tojson }}
+                {%- endif %}
+                {{- '}\n</tool_call>' }}
+            {%- endfor %}
+        {%- endif %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {%- if message.content is string %}
+            {{- message.content }}
+        {%- else %}
+            {%- for content in message.content %}
+                {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
+                    {%- set image_count.value = image_count.value + 1 %}
+                    {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
+                    <|vision_start|><|image_pad|><|vision_end|>
+                {%- elif content.type == 'video' or 'video' in content %}
+                    {%- set video_count.value = video_count.value + 1 %}
+                    {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
+                    <|vision_start|><|video_pad|><|vision_end|>
+                {%- elif 'text' in content %}
+                    {{- content.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {{- '\n</tool_response>' }}
+        {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
+            {{- '<|im_end|>\n' }}
+        {%- endif %}
+    {%- endif %}
+{%- endfor %}
+{%- if add_generation_prompt %}
+    {{- '<|im_start|>assistant\n' }}
+{%- endif %}

tokenizer/tokenizer.json

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

tokenizer/tokenizer_config.json

@@ -0,0 +1,30 @@
+{
+  "add_prefix_space": false,
+  "backend": "tokenizers",
+  "bos_token": null,
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|im_end|>",
+  "errors": "replace",
+  "extra_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "is_local": false,
+  "local_files_only": false,
+  "model_max_length": 262144,
+  "pad_token": "<|endoftext|>",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}

transformer/config.json

@@ -0,0 +1,19 @@
+{
+  "_class_name": "Krea2Transformer2DModel",
+  "_diffusers_version": "0.39.0.dev0",
+  "bias": false,
+  "channels": 16,
+  "features": 6144,
+  "heads": 48,
+  "is_distilled": true,
+  "kvheads": 12,
+  "layers": 28,
+  "multiplier": 4,
+  "patch": 2,
+  "tdim": 256,
+  "theta": 1000.0,
+  "txtdim": 2560,
+  "txtheads": 20,
+  "txtkvheads": 20,
+  "txtlayers": 12
+}

transformer/diffusion_pytorch_model-00001-of-00006.safetensors

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

transformer/diffusion_pytorch_model-00002-of-00006.safetensors

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

transformer/diffusion_pytorch_model-00003-of-00006.safetensors

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

transformer/diffusion_pytorch_model-00004-of-00006.safetensors

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

transformer/diffusion_pytorch_model-00005-of-00006.safetensors

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

transformer/diffusion_pytorch_model-00006-of-00006.safetensors

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

transformer/diffusion_pytorch_model.safetensors.index.json

@@ -0,0 +1,439 @@
+{
+  "metadata": {
+    "total_size": 26585278768
+  },
+  "weight_map": {
+    "blocks.0.attn.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.attn.wk.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.attn.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.attn.wq.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.attn.wv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.mlp.down.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.mlp.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.mlp.up.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.mod.lin": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.postnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.0.prenorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.attn.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.attn.wk.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.attn.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.attn.wq.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.attn.wv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.mlp.down.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.mlp.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.mlp.up.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.mod.lin": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.postnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.1.prenorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.10.attn.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.attn.wk.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.attn.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.attn.wq.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.attn.wv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.mlp.down.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.mlp.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.mlp.up.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.mod.lin": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.postnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.10.prenorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.attn.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.attn.wk.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.attn.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.11.attn.wq.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.attn.wv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.mlp.down.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.11.mlp.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.11.mlp.up.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.11.mod.lin": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.postnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.11.prenorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.12.attn.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.attn.wk.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.attn.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.attn.wq.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.attn.wv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.mlp.down.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.mlp.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.mlp.up.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.mod.lin": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.postnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.12.prenorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.attn.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.attn.wk.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.attn.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.attn.wq.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.attn.wv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.mlp.down.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.mlp.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.mlp.up.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.mod.lin": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.postnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.13.prenorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.attn.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.attn.wk.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.attn.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.attn.wq.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.attn.wv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.mlp.down.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.mlp.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.mlp.up.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.mod.lin": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.postnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.14.prenorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.attn.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.attn.wk.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.attn.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.attn.wq.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.attn.wv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.mlp.down.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.mlp.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.mlp.up.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.mod.lin": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.postnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.15.prenorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.attn.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.attn.wk.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.attn.wo.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.attn.wq.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.attn.wv.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.mlp.down.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.16.mlp.gate.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.mlp.up.weight": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.mod.lin": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.postnorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.16.prenorm.scale": "diffusion_pytorch_model-00003-of-00006.safetensors",
+    "blocks.17.attn.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.attn.wk.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.attn.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.attn.wq.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.attn.wv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.mlp.down.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.mlp.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.mlp.up.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.mod.lin": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.postnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.17.prenorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.attn.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.attn.wk.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.attn.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.attn.wq.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.attn.wv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.mlp.down.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.mlp.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.mlp.up.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.mod.lin": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.postnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.18.prenorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.attn.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.attn.wk.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.attn.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.attn.wq.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.attn.wv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.mlp.down.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.mlp.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.mlp.up.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.mod.lin": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.postnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.19.prenorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.2.attn.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.attn.wk.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.attn.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.attn.wq.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.attn.wv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.mlp.down.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.mlp.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.mlp.up.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.mod.lin": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.postnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.2.prenorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.20.attn.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.attn.wk.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.attn.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.attn.wq.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.attn.wv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.mlp.down.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.mlp.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.mlp.up.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.mod.lin": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.postnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.20.prenorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.attn.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.attn.wk.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.attn.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.attn.wq.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.attn.wv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.mlp.down.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.mlp.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.mlp.up.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.mod.lin": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.postnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.21.prenorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.attn.gate.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.attn.wk.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.attn.wo.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.attn.wq.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.attn.wv.weight": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.mlp.down.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.22.mlp.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.22.mlp.up.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.22.mod.lin": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.postnorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.22.prenorm.scale": "diffusion_pytorch_model-00004-of-00006.safetensors",
+    "blocks.23.attn.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.attn.wk.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.attn.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.attn.wq.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.attn.wv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.mlp.down.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.mlp.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.mlp.up.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.mod.lin": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.postnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.23.prenorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.attn.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.attn.wk.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.attn.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.attn.wq.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.attn.wv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.mlp.down.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.mlp.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.mlp.up.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.mod.lin": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.postnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.24.prenorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.attn.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.attn.wk.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.attn.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.attn.wq.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.attn.wv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.mlp.down.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.mlp.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.mlp.up.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.mod.lin": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.postnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.25.prenorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.attn.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.attn.wk.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.attn.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.attn.wq.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.attn.wv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.mlp.down.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.mlp.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.mlp.up.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.mod.lin": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.postnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.26.prenorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.attn.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.attn.wk.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.attn.wo.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.attn.wq.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.attn.wv.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.mlp.down.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.mlp.gate.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.mlp.up.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.mod.lin": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.postnorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.27.prenorm.scale": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "blocks.3.attn.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.attn.wk.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.attn.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.attn.wq.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.attn.wv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.mlp.down.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.mlp.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.mlp.up.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.mod.lin": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.postnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.3.prenorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.attn.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.attn.wk.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.attn.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.attn.wq.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.attn.wv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.mlp.down.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.mlp.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.mlp.up.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.mod.lin": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.postnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.4.prenorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.attn.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.attn.wk.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.attn.wo.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.attn.wq.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.attn.wv.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.mlp.down.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.5.mlp.gate.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.mlp.up.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.5.mod.lin": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.postnorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.5.prenorm.scale": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "blocks.6.attn.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.attn.wk.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.attn.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.attn.wq.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.attn.wv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.mlp.down.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.mlp.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.mlp.up.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.mod.lin": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.postnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.6.prenorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.attn.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.attn.wk.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.attn.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.attn.wq.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.attn.wv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.mlp.down.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.mlp.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.mlp.up.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.mod.lin": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.postnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.7.prenorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.attn.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.attn.wk.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.attn.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.attn.wq.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.attn.wv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.mlp.down.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.mlp.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.mlp.up.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.mod.lin": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.postnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.8.prenorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.attn.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.attn.wk.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.attn.wo.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.attn.wq.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.attn.wv.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.mlp.down.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.mlp.gate.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.mlp.up.weight": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.mod.lin": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.postnorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "blocks.9.prenorm.scale": "diffusion_pytorch_model-00002-of-00006.safetensors",
+    "first.bias": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "first.weight": "diffusion_pytorch_model-00001-of-00006.safetensors",
+    "last.down.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "last.linear.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "last.linear.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "last.modulation.lin": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "last.norm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "last.up.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "tmlp.0.bias": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "tmlp.0.weight": "diffusion_pytorch_model-00005-of-00006.safetensors",
+    "tmlp.2.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "tmlp.2.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "tproj.1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "tproj.1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.attn.gate.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.attn.wk.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.attn.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.attn.wq.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.attn.wv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.mlp.down.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.mlp.gate.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.mlp.up.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.postnorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.0.prenorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.attn.gate.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.attn.wk.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.attn.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.attn.wq.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.attn.wv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.mlp.down.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.mlp.gate.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.mlp.up.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.postnorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.layerwise_blocks.1.prenorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.projector.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.attn.gate.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.attn.wk.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.attn.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.attn.wq.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.attn.wv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.mlp.down.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.mlp.gate.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.mlp.up.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.postnorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.0.prenorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.attn.gate.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.attn.qknorm.knorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.attn.qknorm.qnorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.attn.wk.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.attn.wo.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.attn.wq.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.attn.wv.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.mlp.down.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.mlp.gate.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.mlp.up.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.postnorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtfusion.refiner_blocks.1.prenorm.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtmlp.0.scale": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtmlp.1.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtmlp.1.weight": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtmlp.3.bias": "diffusion_pytorch_model-00006-of-00006.safetensors",
+    "txtmlp.3.weight": "diffusion_pytorch_model-00006-of-00006.safetensors"
+  }
+}

vae/config.json

@@ -0,0 +1,58 @@
+{
+  "_class_name": "AutoencoderKLQwenImage",
+  "_diffusers_version": "0.39.0.dev0",
+  "_name_or_path": "Qwen/Qwen-Image",
+  "attn_scales": [],
+  "base_dim": 96,
+  "dim_mult": [
+    1,
+    2,
+    4,
+    4
+  ],
+  "dropout": 0.0,
+  "input_channels": 3,
+  "latents_mean": [
+    -0.7571,
+    -0.7089,
+    -0.9113,
+    0.1075,
+    -0.1745,
+    0.9653,
+    -0.1517,
+    1.5508,
+    0.4134,
+    -0.0715,
+    0.5517,
+    -0.3632,
+    -0.1922,
+    -0.9497,
+    0.2503,
+    -0.2921
+  ],
+  "latents_std": [
+    2.8184,
+    1.4541,
+    2.3275,
+    2.6558,
+    1.2196,
+    1.7708,
+    2.6052,
+    2.0743,
+    3.2687,
+    2.1526,
+    2.8652,
+    1.5579,
+    1.6382,
+    1.1253,
+    2.8251,
+    1.916
+  ],
+  "num_res_blocks": 2,
+  "temperal_downsample": [
+    false,
+    true,
+    true
+  ],
+  "z_dim": 16
+}

vae/diffusion_pytorch_model.safetensors

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