SceneWorks commited on
Commit
8dc5d45
·
verified ·
1 Parent(s): 12464ff

Krea 2 Raw MLX turnkey: bf16 / Q8 / Q4 tiers + README + LICENSE (epic 9992 P3, sc-9995)

Browse files
Files changed (40) hide show
  1. .gitattributes +4 -0
  2. LICENSE.pdf +3 -0
  3. README.md +101 -0
  4. bf16/model_index.json +40 -0
  5. bf16/scheduler/scheduler_config.json +18 -0
  6. bf16/text_encoder/config.json +66 -0
  7. bf16/text_encoder/model.safetensors +3 -0
  8. bf16/tokenizer/chat_template.jinja +120 -0
  9. bf16/tokenizer/tokenizer.json +3 -0
  10. bf16/tokenizer/tokenizer_config.json +29 -0
  11. bf16/transformer/config.json +25 -0
  12. bf16/transformer/diffusion_pytorch_model-00001-of-00003.safetensors +3 -0
  13. bf16/transformer/diffusion_pytorch_model-00002-of-00003.safetensors +3 -0
  14. bf16/transformer/diffusion_pytorch_model-00003-of-00003.safetensors +3 -0
  15. bf16/transformer/diffusion_pytorch_model.safetensors.index.json +437 -0
  16. bf16/vae/config.json +58 -0
  17. bf16/vae/diffusion_pytorch_model.safetensors +3 -0
  18. q4/model_index.json +40 -0
  19. q4/scheduler/scheduler_config.json +18 -0
  20. q4/text_encoder/config.json +70 -0
  21. q4/text_encoder/model.safetensors +3 -0
  22. q4/tokenizer/chat_template.jinja +120 -0
  23. q4/tokenizer/tokenizer.json +3 -0
  24. q4/tokenizer/tokenizer_config.json +29 -0
  25. q4/transformer/config.json +29 -0
  26. q4/transformer/diffusion_pytorch_model.safetensors +3 -0
  27. q4/vae/config.json +58 -0
  28. q4/vae/diffusion_pytorch_model.safetensors +3 -0
  29. q8/model_index.json +40 -0
  30. q8/scheduler/scheduler_config.json +18 -0
  31. q8/text_encoder/config.json +70 -0
  32. q8/text_encoder/model.safetensors +3 -0
  33. q8/tokenizer/chat_template.jinja +120 -0
  34. q8/tokenizer/tokenizer.json +3 -0
  35. q8/tokenizer/tokenizer_config.json +29 -0
  36. q8/transformer/config.json +29 -0
  37. q8/transformer/diffusion_pytorch_model.safetensors +3 -0
  38. q8/vae/config.json +58 -0
  39. q8/vae/diffusion_pytorch_model.safetensors +3 -0
  40. upload.log +3 -0
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ LICENSE.pdf filter=lfs diff=lfs merge=lfs -text
37
+ bf16/tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
+ q4/tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
39
+ q8/tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
LICENSE.pdf ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b82a2805162bde714a4eb27b9063c4fc3345d08a30be055134a6160e5430ba74
3
+ size 137711
README.md ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ base_model:
5
+ - krea/Krea-2-Raw
6
+ base_model_relation: quantized
7
+ pipeline_tag: text-to-image
8
+ library_name: mlx
9
+ license: other
10
+ license_name: krea-2-community-license
11
+ license_link: https://huggingface.co/SceneWorks/krea-2-raw-mlx/blob/main/LICENSE.pdf
12
+ tags:
13
+ - mlx
14
+ - apple-silicon
15
+ - text-to-image
16
+ - diffusion
17
+ - krea-2
18
+ - quantized
19
+ ---
20
+
21
+ # Krea 2 Raw — MLX (turnkey: bf16 / Q8 / Q4)
22
+
23
+ On-device, Apple-MLX-ready repack of **[krea/Krea-2-Raw](https://huggingface.co/krea/Krea-2-Raw)**, the
24
+ **undistilled** 12B single-stream text-to-image checkpoint from **Krea.ai, Inc.** — the full
25
+ classifier-free-guidance base model (and the LoRA-training base) behind the distilled Krea 2 Turbo.
26
+ This repository is a **Derivative** prepared for [`mlx-gen`](https://github.com/michaeltrefry/mlx-gen)
27
+ (and the SceneWorks worker that embeds it): the weights are group-wise-affine **quantized and repacked**
28
+ from the original bf16 diffusers checkpoint so the model loads and runs natively on Apple Silicon with no
29
+ Python/PyTorch sidecar. The `bf16/` tier is the dense original re-layout (max fidelity + the LoRA-training
30
+ base).
31
+
32
+ This is **not** the original checkpoint. For the reference model, training details, and the canonical
33
+ diffusers / SGLang inference paths, see the upstream card: **<https://huggingface.co/krea/Krea-2-Raw>**.
34
+
35
+ ## Attribution
36
+
37
+ - **Original model:** Krea 2 Raw — © **Krea.ai, Inc.**, released 2026-06-22.
38
+ - **Base model:** [`krea/Krea-2-Raw`](https://huggingface.co/krea/Krea-2-Raw) (the undistilled base; Krea 2
39
+ Turbo is distilled from it).
40
+ - **This Derivative:** quantized + MLX-repacked by the SceneWorks / `mlx-gen` project. No retraining or
41
+ fine-tuning was performed — only numerical quantization and on-disk re-layout.
42
+
43
+ ## License
44
+
45
+ Use of these weights is governed by the **Krea 2 Community License Agreement** and the Krea Acceptable Use
46
+ Policy, exactly as for the original model. A copy of the license is included in this repository as
47
+ [`LICENSE.pdf`](LICENSE.pdf) (also at
48
+ <https://huggingface.co/krea/Krea-2-Raw/blob/main/LICENSE.pdf>). In the event of any conflict, the Krea
49
+ Acceptable Use Policy and Krea 2 Community License control.
50
+
51
+ > **Deployer obligation (content filtering).** The Krea 2 Community License requires anyone who deploys the
52
+ > model to implement content-filtering measures or equivalent review processes appropriate to their use
53
+ > case, to prevent the generation or distribution of unlawful or policy-violating content. If you serve
54
+ > this model, you are responsible for those safeguards. Report harmful, illegal, or policy-violating
55
+ > outputs to **safety@krea.ai** (potential CSAM is escalated to NCMEC as required by law).
56
+
57
+ Krea does not claim copyright over generated outputs; users are solely responsible for their inputs and any
58
+ use of the outputs.
59
+
60
+ ## What changed vs. the upstream checkpoint
61
+
62
+ The conversion is **lossy only through quantization** — the architecture, tokenizer, scheduler config, and
63
+ VAE are byte-for-byte the originals.
64
+
65
+ - **Transformer (DiT)** and **Qwen3-VL-4B text encoder**: for the Q8 / Q4 tiers the linear projection
66
+ weights are quantized to **group-wise affine Q8 / Q4** (group size 64) and repacked into a single
67
+ `.safetensors` per stack. Norms, embeddings, modulation tables, and the text-encoder vision tower stay
68
+ dense. The `bf16/` tier keeps every weight dense.
69
+ - **VAE** (`AutoencoderKLQwenImage`): copied **unchanged** (f32).
70
+ - **`tokenizer/`, `scheduler/`, `model_index.json`**: copied unchanged.
71
+
72
+ ## Repository layout
73
+
74
+ Each tier is a complete, self-contained snapshot you can load directly:
75
+
76
+ | Path | Quantization | On-disk size | Notes |
77
+ |--------|--------------------|--------------|--------------------------------------------------------------|
78
+ | `bf16/`| none (dense bf16) | ~35.7 GB | Max fidelity; the LoRA-training base. |
79
+ | `q8/` | Q8 (group size 64) | ~20.6 GB | **Default.** Near-lossless; needs a 48 GB-class Mac. |
80
+ | `q4/` | Q4 (group size 64) | ~12.5 GB | Lighter footprint; mild quality trade-off. |
81
+
82
+ ```
83
+ krea-2-raw-mlx/
84
+ ├── LICENSE.pdf
85
+ ├── README.md
86
+ ├── bf16/ { transformer/ text_encoder/ vae/ tokenizer/ scheduler/ model_index.json }
87
+ ├── q8/ { transformer/ text_encoder/ vae/ tokenizer/ scheduler/ model_index.json }
88
+ └── q4/ { transformer/ text_encoder/ vae/ tokenizer/ scheduler/ model_index.json }
89
+ ```
90
+
91
+ ## Usage
92
+
93
+ Built for Apple-Silicon inference through `mlx-gen`'s `krea_2_raw` engine. Point a loader at a tier
94
+ subdirectory (`bf16/`, `q8/`, or `q4/`); it auto-detects the packed weights. Unlike the CFG-free Turbo,
95
+ Krea 2 Raw is a **true classifier-free-guidance** model — run ~52 steps with a real guidance scale (~3.5)
96
+ and an optional negative prompt. The same `bf16/` tier is also the base for Krea 2 LoRA training.
97
+
98
+ ## Model details
99
+
100
+ See the upstream card for the full model overview, capabilities, intended/out-of-scope uses, training-data
101
+ summary, safety measures, and risk/limitation disclosures: <https://huggingface.co/krea/Krea-2-Raw>.
bf16/model_index.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Krea2Pipeline",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "FlowMatchEulerDiscreteScheduler"
7
+ ],
8
+ "text_encoder": [
9
+ "transformers",
10
+ "Qwen3VLModel"
11
+ ],
12
+ "text_encoder_select_layers": [
13
+ 2,
14
+ 5,
15
+ 8,
16
+ 11,
17
+ 14,
18
+ 17,
19
+ 20,
20
+ 23,
21
+ 26,
22
+ 29,
23
+ 32,
24
+ 35
25
+ ],
26
+ "tokenizer": [
27
+ "transformers",
28
+ "Qwen2Tokenizer"
29
+ ],
30
+ "transformer": [
31
+ "diffusers",
32
+ "Krea2Transformer2DModel"
33
+ ],
34
+ "vae": [
35
+ "diffusers",
36
+ "AutoencoderKLQwenImage"
37
+ ],
38
+ "is_distilled": false,
39
+ "patch_size": 2
40
+ }
bf16/scheduler/scheduler_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "FlowMatchEulerDiscreteScheduler",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "base_image_seq_len": 256,
5
+ "base_shift": 0.5,
6
+ "invert_sigmas": false,
7
+ "max_image_seq_len": 6400,
8
+ "max_shift": 1.15,
9
+ "num_train_timesteps": 1000,
10
+ "shift": 1.0,
11
+ "shift_terminal": null,
12
+ "stochastic_sampling": false,
13
+ "time_shift_type": "exponential",
14
+ "use_beta_sigmas": false,
15
+ "use_dynamic_shifting": true,
16
+ "use_exponential_sigmas": false,
17
+ "use_karras_sigmas": false
18
+ }
bf16/text_encoder/config.json ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3VLModel"
4
+ ],
5
+ "dtype": "bfloat16",
6
+ "image_token_id": 151655,
7
+ "model_type": "qwen3_vl",
8
+ "text_config": {
9
+ "attention_bias": false,
10
+ "attention_dropout": 0.0,
11
+ "bos_token_id": 151643,
12
+ "dtype": "bfloat16",
13
+ "eos_token_id": 151645,
14
+ "head_dim": 128,
15
+ "hidden_act": "silu",
16
+ "hidden_size": 2560,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 9728,
19
+ "max_position_embeddings": 262144,
20
+ "model_type": "qwen3_vl_text",
21
+ "num_attention_heads": 32,
22
+ "num_hidden_layers": 36,
23
+ "num_key_value_heads": 8,
24
+ "pad_token_id": null,
25
+ "rms_norm_eps": 1e-06,
26
+ "rope_parameters": {
27
+ "mrope_interleaved": true,
28
+ "mrope_section": [
29
+ 24,
30
+ 20,
31
+ 20
32
+ ],
33
+ "rope_theta": 5000000,
34
+ "rope_type": "default"
35
+ },
36
+ "tie_word_embeddings": true,
37
+ "use_cache": true,
38
+ "vocab_size": 151936
39
+ },
40
+ "tie_word_embeddings": true,
41
+ "transformers_version": "5.2.0",
42
+ "video_token_id": 151656,
43
+ "vision_config": {
44
+ "deepstack_visual_indexes": [
45
+ 5,
46
+ 11,
47
+ 17
48
+ ],
49
+ "depth": 24,
50
+ "dtype": "bfloat16",
51
+ "hidden_act": "gelu_pytorch_tanh",
52
+ "hidden_size": 1024,
53
+ "in_channels": 3,
54
+ "initializer_range": 0.02,
55
+ "intermediate_size": 4096,
56
+ "model_type": "qwen3_vl",
57
+ "num_heads": 16,
58
+ "num_position_embeddings": 2304,
59
+ "out_hidden_size": 2560,
60
+ "patch_size": 16,
61
+ "spatial_merge_size": 2,
62
+ "temporal_patch_size": 2
63
+ },
64
+ "vision_end_token_id": 151653,
65
+ "vision_start_token_id": 151652
66
+ }
bf16/text_encoder/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8434db05292f95e0041589a7c82abeb39385be59c85b54ae11caa7b45e9f4f13
3
+ size 8875715136
bf16/tokenizer/chat_template.jinja ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0].role == 'system' %}
4
+ {%- if messages[0].content is string %}
5
+ {{- messages[0].content }}
6
+ {%- else %}
7
+ {%- for content in messages[0].content %}
8
+ {%- if 'text' in content %}
9
+ {{- content.text }}
10
+ {%- endif %}
11
+ {%- endfor %}
12
+ {%- endif %}
13
+ {{- '\n\n' }}
14
+ {%- endif %}
15
+ {{- "# 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>" }}
16
+ {%- for tool in tools %}
17
+ {{- "\n" }}
18
+ {{- tool | tojson }}
19
+ {%- endfor %}
20
+ {{- "\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" }}
21
+ {%- else %}
22
+ {%- if messages[0].role == 'system' %}
23
+ {{- '<|im_start|>system\n' }}
24
+ {%- if messages[0].content is string %}
25
+ {{- messages[0].content }}
26
+ {%- else %}
27
+ {%- for content in messages[0].content %}
28
+ {%- if 'text' in content %}
29
+ {{- content.text }}
30
+ {%- endif %}
31
+ {%- endfor %}
32
+ {%- endif %}
33
+ {{- '<|im_end|>\n' }}
34
+ {%- endif %}
35
+ {%- endif %}
36
+ {%- set image_count = namespace(value=0) %}
37
+ {%- set video_count = namespace(value=0) %}
38
+ {%- for message in messages %}
39
+ {%- if message.role == "user" %}
40
+ {{- '<|im_start|>' + message.role + '\n' }}
41
+ {%- if message.content is string %}
42
+ {{- message.content }}
43
+ {%- else %}
44
+ {%- for content in message.content %}
45
+ {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
46
+ {%- set image_count.value = image_count.value + 1 %}
47
+ {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
48
+ <|vision_start|><|image_pad|><|vision_end|>
49
+ {%- elif content.type == 'video' or 'video' in content %}
50
+ {%- set video_count.value = video_count.value + 1 %}
51
+ {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
52
+ <|vision_start|><|video_pad|><|vision_end|>
53
+ {%- elif 'text' in content %}
54
+ {{- content.text }}
55
+ {%- endif %}
56
+ {%- endfor %}
57
+ {%- endif %}
58
+ {{- '<|im_end|>\n' }}
59
+ {%- elif message.role == "assistant" %}
60
+ {{- '<|im_start|>' + message.role + '\n' }}
61
+ {%- if message.content is string %}
62
+ {{- message.content }}
63
+ {%- else %}
64
+ {%- for content_item in message.content %}
65
+ {%- if 'text' in content_item %}
66
+ {{- content_item.text }}
67
+ {%- endif %}
68
+ {%- endfor %}
69
+ {%- endif %}
70
+ {%- if message.tool_calls %}
71
+ {%- for tool_call in message.tool_calls %}
72
+ {%- if (loop.first and message.content) or (not loop.first) %}
73
+ {{- '\n' }}
74
+ {%- endif %}
75
+ {%- if tool_call.function %}
76
+ {%- set tool_call = tool_call.function %}
77
+ {%- endif %}
78
+ {{- '<tool_call>\n{"name": "' }}
79
+ {{- tool_call.name }}
80
+ {{- '", "arguments": ' }}
81
+ {%- if tool_call.arguments is string %}
82
+ {{- tool_call.arguments }}
83
+ {%- else %}
84
+ {{- tool_call.arguments | tojson }}
85
+ {%- endif %}
86
+ {{- '}\n</tool_call>' }}
87
+ {%- endfor %}
88
+ {%- endif %}
89
+ {{- '<|im_end|>\n' }}
90
+ {%- elif message.role == "tool" %}
91
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
92
+ {{- '<|im_start|>user' }}
93
+ {%- endif %}
94
+ {{- '\n<tool_response>\n' }}
95
+ {%- if message.content is string %}
96
+ {{- message.content }}
97
+ {%- else %}
98
+ {%- for content in message.content %}
99
+ {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
100
+ {%- set image_count.value = image_count.value + 1 %}
101
+ {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
102
+ <|vision_start|><|image_pad|><|vision_end|>
103
+ {%- elif content.type == 'video' or 'video' in content %}
104
+ {%- set video_count.value = video_count.value + 1 %}
105
+ {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
106
+ <|vision_start|><|video_pad|><|vision_end|>
107
+ {%- elif 'text' in content %}
108
+ {{- content.text }}
109
+ {%- endif %}
110
+ {%- endfor %}
111
+ {%- endif %}
112
+ {{- '\n</tool_response>' }}
113
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
114
+ {{- '<|im_end|>\n' }}
115
+ {%- endif %}
116
+ {%- endif %}
117
+ {%- endfor %}
118
+ {%- if add_generation_prompt %}
119
+ {{- '<|im_start|>assistant\n' }}
120
+ {%- endif %}
bf16/tokenizer/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be75606093db2094d7cd20f3c2f385c212750648bd6ea4fb2bf507a6a4c55506
3
+ size 11422650
bf16/tokenizer/tokenizer_config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "extra_special_tokens": [
9
+ "<|im_start|>",
10
+ "<|im_end|>",
11
+ "<|object_ref_start|>",
12
+ "<|object_ref_end|>",
13
+ "<|box_start|>",
14
+ "<|box_end|>",
15
+ "<|quad_start|>",
16
+ "<|quad_end|>",
17
+ "<|vision_start|>",
18
+ "<|vision_end|>",
19
+ "<|vision_pad|>",
20
+ "<|image_pad|>",
21
+ "<|video_pad|>"
22
+ ],
23
+ "is_local": true,
24
+ "model_max_length": 262144,
25
+ "pad_token": "<|endoftext|>",
26
+ "split_special_tokens": false,
27
+ "tokenizer_class": "Qwen2Tokenizer",
28
+ "unk_token": null
29
+ }
bf16/transformer/config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Krea2Transformer2DModel",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "attention_head_dim": 128,
5
+ "axes_dims_rope": [
6
+ 32,
7
+ 48,
8
+ 48
9
+ ],
10
+ "in_channels": 64,
11
+ "intermediate_size": 16384,
12
+ "norm_eps": 1e-05,
13
+ "num_attention_heads": 48,
14
+ "num_key_value_heads": 12,
15
+ "num_layers": 28,
16
+ "num_layerwise_text_blocks": 2,
17
+ "num_refiner_text_blocks": 2,
18
+ "num_text_layers": 12,
19
+ "rope_theta": 1000.0,
20
+ "text_hidden_dim": 2560,
21
+ "text_intermediate_size": 6912,
22
+ "text_num_attention_heads": 20,
23
+ "text_num_key_value_heads": 20,
24
+ "timestep_embed_dim": 256
25
+ }
bf16/transformer/diffusion_pytorch_model-00001-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aeed04b3fd933ff35ffb898548a585a32f2146ee6681700e2b50e4478d52d58a
3
+ size 9970652128
bf16/transformer/diffusion_pytorch_model-00002-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e606797c9fbfeb820b9a0222823915ac1459d11a27caa84d53aba150c85f2b26
3
+ size 9829444456
bf16/transformer/diffusion_pytorch_model-00003-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9300cac5c4cdbc33bbccadf176c0e5a8314b6a4aabefae7a2781782d50d78e7d
3
+ size 6483240784
bf16/transformer/diffusion_pytorch_model.safetensors.index.json ADDED
@@ -0,0 +1,437 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 26283288880
4
+ },
5
+ "weight_map": {
6
+ "final_layer.linear.bias": "diffusion_pytorch_model-00003-of-00003.safetensors",
7
+ "final_layer.linear.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
8
+ "final_layer.norm.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
9
+ "final_layer.scale_shift_table": "diffusion_pytorch_model-00003-of-00003.safetensors",
10
+ "img_in.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
11
+ "img_in.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
12
+ "text_fusion.layerwise_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
13
+ "text_fusion.layerwise_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
14
+ "text_fusion.layerwise_blocks.0.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
15
+ "text_fusion.layerwise_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
16
+ "text_fusion.layerwise_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
17
+ "text_fusion.layerwise_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
18
+ "text_fusion.layerwise_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
19
+ "text_fusion.layerwise_blocks.0.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
20
+ "text_fusion.layerwise_blocks.0.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
21
+ "text_fusion.layerwise_blocks.0.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
22
+ "text_fusion.layerwise_blocks.0.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
23
+ "text_fusion.layerwise_blocks.0.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
24
+ "text_fusion.layerwise_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
25
+ "text_fusion.layerwise_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
26
+ "text_fusion.layerwise_blocks.1.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
27
+ "text_fusion.layerwise_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
28
+ "text_fusion.layerwise_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
29
+ "text_fusion.layerwise_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
30
+ "text_fusion.layerwise_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
31
+ "text_fusion.layerwise_blocks.1.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
32
+ "text_fusion.layerwise_blocks.1.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
33
+ "text_fusion.layerwise_blocks.1.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
34
+ "text_fusion.layerwise_blocks.1.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
35
+ "text_fusion.layerwise_blocks.1.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
36
+ "text_fusion.projector.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
37
+ "text_fusion.refiner_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
38
+ "text_fusion.refiner_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
39
+ "text_fusion.refiner_blocks.0.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
40
+ "text_fusion.refiner_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
41
+ "text_fusion.refiner_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
42
+ "text_fusion.refiner_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
43
+ "text_fusion.refiner_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
44
+ "text_fusion.refiner_blocks.0.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
45
+ "text_fusion.refiner_blocks.0.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
46
+ "text_fusion.refiner_blocks.0.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
47
+ "text_fusion.refiner_blocks.0.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
48
+ "text_fusion.refiner_blocks.0.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
49
+ "text_fusion.refiner_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
50
+ "text_fusion.refiner_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
51
+ "text_fusion.refiner_blocks.1.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
52
+ "text_fusion.refiner_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
53
+ "text_fusion.refiner_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
54
+ "text_fusion.refiner_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
55
+ "text_fusion.refiner_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
56
+ "text_fusion.refiner_blocks.1.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
57
+ "text_fusion.refiner_blocks.1.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
58
+ "text_fusion.refiner_blocks.1.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
59
+ "text_fusion.refiner_blocks.1.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
60
+ "text_fusion.refiner_blocks.1.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
61
+ "time_embed.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
62
+ "time_embed.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
63
+ "time_embed.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
64
+ "time_embed.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
65
+ "time_mod_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
66
+ "time_mod_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
67
+ "transformer_blocks.0.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
68
+ "transformer_blocks.0.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
69
+ "transformer_blocks.0.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
70
+ "transformer_blocks.0.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
71
+ "transformer_blocks.0.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
72
+ "transformer_blocks.0.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
73
+ "transformer_blocks.0.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
74
+ "transformer_blocks.0.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
75
+ "transformer_blocks.0.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
76
+ "transformer_blocks.0.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
77
+ "transformer_blocks.0.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
78
+ "transformer_blocks.0.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
79
+ "transformer_blocks.0.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
80
+ "transformer_blocks.1.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
81
+ "transformer_blocks.1.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
82
+ "transformer_blocks.1.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
83
+ "transformer_blocks.1.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
84
+ "transformer_blocks.1.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
85
+ "transformer_blocks.1.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
86
+ "transformer_blocks.1.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
87
+ "transformer_blocks.1.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
88
+ "transformer_blocks.1.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
89
+ "transformer_blocks.1.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
90
+ "transformer_blocks.1.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
91
+ "transformer_blocks.1.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
92
+ "transformer_blocks.1.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
93
+ "transformer_blocks.10.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
94
+ "transformer_blocks.10.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
95
+ "transformer_blocks.10.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
96
+ "transformer_blocks.10.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
97
+ "transformer_blocks.10.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
98
+ "transformer_blocks.10.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
99
+ "transformer_blocks.10.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
100
+ "transformer_blocks.10.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
101
+ "transformer_blocks.10.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
102
+ "transformer_blocks.10.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
103
+ "transformer_blocks.10.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
104
+ "transformer_blocks.10.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
105
+ "transformer_blocks.10.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
106
+ "transformer_blocks.11.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
107
+ "transformer_blocks.11.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
108
+ "transformer_blocks.11.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
109
+ "transformer_blocks.11.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
110
+ "transformer_blocks.11.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
111
+ "transformer_blocks.11.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
112
+ "transformer_blocks.11.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
113
+ "transformer_blocks.11.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
114
+ "transformer_blocks.11.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
115
+ "transformer_blocks.11.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
116
+ "transformer_blocks.11.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
117
+ "transformer_blocks.11.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
118
+ "transformer_blocks.11.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
119
+ "transformer_blocks.12.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
120
+ "transformer_blocks.12.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
121
+ "transformer_blocks.12.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
122
+ "transformer_blocks.12.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
123
+ "transformer_blocks.12.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
124
+ "transformer_blocks.12.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
125
+ "transformer_blocks.12.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
126
+ "transformer_blocks.12.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
127
+ "transformer_blocks.12.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
128
+ "transformer_blocks.12.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
129
+ "transformer_blocks.12.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
130
+ "transformer_blocks.12.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
131
+ "transformer_blocks.12.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
132
+ "transformer_blocks.13.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
133
+ "transformer_blocks.13.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
134
+ "transformer_blocks.13.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
135
+ "transformer_blocks.13.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
136
+ "transformer_blocks.13.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
137
+ "transformer_blocks.13.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
138
+ "transformer_blocks.13.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
139
+ "transformer_blocks.13.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
140
+ "transformer_blocks.13.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
141
+ "transformer_blocks.13.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
142
+ "transformer_blocks.13.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
143
+ "transformer_blocks.13.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
144
+ "transformer_blocks.13.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
145
+ "transformer_blocks.14.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
146
+ "transformer_blocks.14.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
147
+ "transformer_blocks.14.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
148
+ "transformer_blocks.14.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
149
+ "transformer_blocks.14.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
150
+ "transformer_blocks.14.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
151
+ "transformer_blocks.14.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
152
+ "transformer_blocks.14.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
153
+ "transformer_blocks.14.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
154
+ "transformer_blocks.14.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
155
+ "transformer_blocks.14.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
156
+ "transformer_blocks.14.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
157
+ "transformer_blocks.14.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
158
+ "transformer_blocks.15.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
159
+ "transformer_blocks.15.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
160
+ "transformer_blocks.15.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
161
+ "transformer_blocks.15.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
162
+ "transformer_blocks.15.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
163
+ "transformer_blocks.15.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
164
+ "transformer_blocks.15.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
165
+ "transformer_blocks.15.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
166
+ "transformer_blocks.15.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
167
+ "transformer_blocks.15.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
168
+ "transformer_blocks.15.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
169
+ "transformer_blocks.15.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
170
+ "transformer_blocks.15.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
171
+ "transformer_blocks.16.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
172
+ "transformer_blocks.16.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
173
+ "transformer_blocks.16.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
174
+ "transformer_blocks.16.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
175
+ "transformer_blocks.16.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
176
+ "transformer_blocks.16.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
177
+ "transformer_blocks.16.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
178
+ "transformer_blocks.16.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
179
+ "transformer_blocks.16.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
180
+ "transformer_blocks.16.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
181
+ "transformer_blocks.16.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
182
+ "transformer_blocks.16.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
183
+ "transformer_blocks.16.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
184
+ "transformer_blocks.17.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
185
+ "transformer_blocks.17.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
186
+ "transformer_blocks.17.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
187
+ "transformer_blocks.17.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
188
+ "transformer_blocks.17.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
189
+ "transformer_blocks.17.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
190
+ "transformer_blocks.17.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
191
+ "transformer_blocks.17.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
192
+ "transformer_blocks.17.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
193
+ "transformer_blocks.17.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
194
+ "transformer_blocks.17.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
195
+ "transformer_blocks.17.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
196
+ "transformer_blocks.17.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
197
+ "transformer_blocks.18.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
198
+ "transformer_blocks.18.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
199
+ "transformer_blocks.18.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
200
+ "transformer_blocks.18.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
201
+ "transformer_blocks.18.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
202
+ "transformer_blocks.18.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
203
+ "transformer_blocks.18.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
204
+ "transformer_blocks.18.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
205
+ "transformer_blocks.18.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
206
+ "transformer_blocks.18.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
207
+ "transformer_blocks.18.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
208
+ "transformer_blocks.18.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
209
+ "transformer_blocks.18.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
210
+ "transformer_blocks.19.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
211
+ "transformer_blocks.19.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
212
+ "transformer_blocks.19.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
213
+ "transformer_blocks.19.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
214
+ "transformer_blocks.19.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
215
+ "transformer_blocks.19.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
216
+ "transformer_blocks.19.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
217
+ "transformer_blocks.19.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
218
+ "transformer_blocks.19.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
219
+ "transformer_blocks.19.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
220
+ "transformer_blocks.19.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
221
+ "transformer_blocks.19.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
222
+ "transformer_blocks.19.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
223
+ "transformer_blocks.2.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
224
+ "transformer_blocks.2.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
225
+ "transformer_blocks.2.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
226
+ "transformer_blocks.2.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
227
+ "transformer_blocks.2.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
228
+ "transformer_blocks.2.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
229
+ "transformer_blocks.2.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
230
+ "transformer_blocks.2.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
231
+ "transformer_blocks.2.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
232
+ "transformer_blocks.2.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
233
+ "transformer_blocks.2.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
234
+ "transformer_blocks.2.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
235
+ "transformer_blocks.2.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
236
+ "transformer_blocks.20.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
237
+ "transformer_blocks.20.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
238
+ "transformer_blocks.20.attn.to_gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
239
+ "transformer_blocks.20.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
240
+ "transformer_blocks.20.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
241
+ "transformer_blocks.20.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
242
+ "transformer_blocks.20.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
243
+ "transformer_blocks.20.ff.down.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
244
+ "transformer_blocks.20.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
245
+ "transformer_blocks.20.ff.up.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
246
+ "transformer_blocks.20.norm1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
247
+ "transformer_blocks.20.norm2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
248
+ "transformer_blocks.20.scale_shift_table": "diffusion_pytorch_model-00002-of-00003.safetensors",
249
+ "transformer_blocks.21.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
250
+ "transformer_blocks.21.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
251
+ "transformer_blocks.21.attn.to_gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
252
+ "transformer_blocks.21.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
253
+ "transformer_blocks.21.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
254
+ "transformer_blocks.21.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
255
+ "transformer_blocks.21.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
256
+ "transformer_blocks.21.ff.down.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
257
+ "transformer_blocks.21.ff.gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
258
+ "transformer_blocks.21.ff.up.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
259
+ "transformer_blocks.21.norm1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
260
+ "transformer_blocks.21.norm2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
261
+ "transformer_blocks.21.scale_shift_table": "diffusion_pytorch_model-00003-of-00003.safetensors",
262
+ "transformer_blocks.22.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
263
+ "transformer_blocks.22.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
264
+ "transformer_blocks.22.attn.to_gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
265
+ "transformer_blocks.22.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
266
+ "transformer_blocks.22.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
267
+ "transformer_blocks.22.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
268
+ "transformer_blocks.22.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
269
+ "transformer_blocks.22.ff.down.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
270
+ "transformer_blocks.22.ff.gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
271
+ "transformer_blocks.22.ff.up.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
272
+ "transformer_blocks.22.norm1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
273
+ "transformer_blocks.22.norm2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
274
+ "transformer_blocks.22.scale_shift_table": "diffusion_pytorch_model-00003-of-00003.safetensors",
275
+ "transformer_blocks.23.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
276
+ "transformer_blocks.23.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
277
+ "transformer_blocks.23.attn.to_gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
278
+ "transformer_blocks.23.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
279
+ "transformer_blocks.23.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
280
+ "transformer_blocks.23.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
281
+ "transformer_blocks.23.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
282
+ "transformer_blocks.23.ff.down.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
283
+ "transformer_blocks.23.ff.gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
284
+ "transformer_blocks.23.ff.up.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
285
+ "transformer_blocks.23.norm1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
286
+ "transformer_blocks.23.norm2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
287
+ "transformer_blocks.23.scale_shift_table": "diffusion_pytorch_model-00003-of-00003.safetensors",
288
+ "transformer_blocks.24.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
289
+ "transformer_blocks.24.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
290
+ "transformer_blocks.24.attn.to_gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
291
+ "transformer_blocks.24.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
292
+ "transformer_blocks.24.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
293
+ "transformer_blocks.24.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
294
+ "transformer_blocks.24.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
295
+ "transformer_blocks.24.ff.down.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
296
+ "transformer_blocks.24.ff.gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
297
+ "transformer_blocks.24.ff.up.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
298
+ "transformer_blocks.24.norm1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
299
+ "transformer_blocks.24.norm2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
300
+ "transformer_blocks.24.scale_shift_table": "diffusion_pytorch_model-00003-of-00003.safetensors",
301
+ "transformer_blocks.25.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
302
+ "transformer_blocks.25.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
303
+ "transformer_blocks.25.attn.to_gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
304
+ "transformer_blocks.25.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
305
+ "transformer_blocks.25.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
306
+ "transformer_blocks.25.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
307
+ "transformer_blocks.25.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
308
+ "transformer_blocks.25.ff.down.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
309
+ "transformer_blocks.25.ff.gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
310
+ "transformer_blocks.25.ff.up.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
311
+ "transformer_blocks.25.norm1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
312
+ "transformer_blocks.25.norm2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
313
+ "transformer_blocks.25.scale_shift_table": "diffusion_pytorch_model-00003-of-00003.safetensors",
314
+ "transformer_blocks.26.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
315
+ "transformer_blocks.26.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
316
+ "transformer_blocks.26.attn.to_gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
317
+ "transformer_blocks.26.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
318
+ "transformer_blocks.26.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
319
+ "transformer_blocks.26.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
320
+ "transformer_blocks.26.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
321
+ "transformer_blocks.26.ff.down.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
322
+ "transformer_blocks.26.ff.gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
323
+ "transformer_blocks.26.ff.up.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
324
+ "transformer_blocks.26.norm1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
325
+ "transformer_blocks.26.norm2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
326
+ "transformer_blocks.26.scale_shift_table": "diffusion_pytorch_model-00003-of-00003.safetensors",
327
+ "transformer_blocks.27.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
328
+ "transformer_blocks.27.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
329
+ "transformer_blocks.27.attn.to_gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
330
+ "transformer_blocks.27.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
331
+ "transformer_blocks.27.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
332
+ "transformer_blocks.27.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
333
+ "transformer_blocks.27.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
334
+ "transformer_blocks.27.ff.down.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
335
+ "transformer_blocks.27.ff.gate.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
336
+ "transformer_blocks.27.ff.up.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
337
+ "transformer_blocks.27.norm1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
338
+ "transformer_blocks.27.norm2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors",
339
+ "transformer_blocks.27.scale_shift_table": "diffusion_pytorch_model-00003-of-00003.safetensors",
340
+ "transformer_blocks.3.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
341
+ "transformer_blocks.3.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
342
+ "transformer_blocks.3.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
343
+ "transformer_blocks.3.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
344
+ "transformer_blocks.3.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
345
+ "transformer_blocks.3.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
346
+ "transformer_blocks.3.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
347
+ "transformer_blocks.3.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
348
+ "transformer_blocks.3.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
349
+ "transformer_blocks.3.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
350
+ "transformer_blocks.3.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
351
+ "transformer_blocks.3.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
352
+ "transformer_blocks.3.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
353
+ "transformer_blocks.4.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
354
+ "transformer_blocks.4.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
355
+ "transformer_blocks.4.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
356
+ "transformer_blocks.4.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
357
+ "transformer_blocks.4.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
358
+ "transformer_blocks.4.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
359
+ "transformer_blocks.4.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
360
+ "transformer_blocks.4.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
361
+ "transformer_blocks.4.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
362
+ "transformer_blocks.4.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
363
+ "transformer_blocks.4.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
364
+ "transformer_blocks.4.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
365
+ "transformer_blocks.4.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
366
+ "transformer_blocks.5.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
367
+ "transformer_blocks.5.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
368
+ "transformer_blocks.5.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
369
+ "transformer_blocks.5.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
370
+ "transformer_blocks.5.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
371
+ "transformer_blocks.5.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
372
+ "transformer_blocks.5.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
373
+ "transformer_blocks.5.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
374
+ "transformer_blocks.5.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
375
+ "transformer_blocks.5.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
376
+ "transformer_blocks.5.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
377
+ "transformer_blocks.5.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
378
+ "transformer_blocks.5.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
379
+ "transformer_blocks.6.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
380
+ "transformer_blocks.6.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
381
+ "transformer_blocks.6.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
382
+ "transformer_blocks.6.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
383
+ "transformer_blocks.6.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
384
+ "transformer_blocks.6.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
385
+ "transformer_blocks.6.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
386
+ "transformer_blocks.6.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
387
+ "transformer_blocks.6.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
388
+ "transformer_blocks.6.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
389
+ "transformer_blocks.6.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
390
+ "transformer_blocks.6.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
391
+ "transformer_blocks.6.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
392
+ "transformer_blocks.7.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
393
+ "transformer_blocks.7.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
394
+ "transformer_blocks.7.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
395
+ "transformer_blocks.7.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
396
+ "transformer_blocks.7.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
397
+ "transformer_blocks.7.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
398
+ "transformer_blocks.7.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
399
+ "transformer_blocks.7.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
400
+ "transformer_blocks.7.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
401
+ "transformer_blocks.7.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
402
+ "transformer_blocks.7.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
403
+ "transformer_blocks.7.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
404
+ "transformer_blocks.7.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
405
+ "transformer_blocks.8.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
406
+ "transformer_blocks.8.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
407
+ "transformer_blocks.8.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
408
+ "transformer_blocks.8.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
409
+ "transformer_blocks.8.attn.to_out.0.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
410
+ "transformer_blocks.8.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
411
+ "transformer_blocks.8.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
412
+ "transformer_blocks.8.ff.down.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
413
+ "transformer_blocks.8.ff.gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
414
+ "transformer_blocks.8.ff.up.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
415
+ "transformer_blocks.8.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
416
+ "transformer_blocks.8.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
417
+ "transformer_blocks.8.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
418
+ "transformer_blocks.9.attn.norm_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
419
+ "transformer_blocks.9.attn.norm_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
420
+ "transformer_blocks.9.attn.to_gate.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
421
+ "transformer_blocks.9.attn.to_k.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
422
+ "transformer_blocks.9.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
423
+ "transformer_blocks.9.attn.to_q.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
424
+ "transformer_blocks.9.attn.to_v.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
425
+ "transformer_blocks.9.ff.down.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
426
+ "transformer_blocks.9.ff.gate.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
427
+ "transformer_blocks.9.ff.up.weight": "diffusion_pytorch_model-00002-of-00003.safetensors",
428
+ "transformer_blocks.9.norm1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
429
+ "transformer_blocks.9.norm2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
430
+ "transformer_blocks.9.scale_shift_table": "diffusion_pytorch_model-00001-of-00003.safetensors",
431
+ "txt_in.linear_1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
432
+ "txt_in.linear_1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
433
+ "txt_in.linear_2.bias": "diffusion_pytorch_model-00001-of-00003.safetensors",
434
+ "txt_in.linear_2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors",
435
+ "txt_in.norm.weight": "diffusion_pytorch_model-00001-of-00003.safetensors"
436
+ }
437
+ }
bf16/vae/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKLQwenImage",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "_name_or_path": "Qwen/Qwen-Image",
5
+ "attn_scales": [],
6
+ "base_dim": 96,
7
+ "dim_mult": [
8
+ 1,
9
+ 2,
10
+ 4,
11
+ 4
12
+ ],
13
+ "dropout": 0.0,
14
+ "input_channels": 3,
15
+ "latents_mean": [
16
+ -0.7571,
17
+ -0.7089,
18
+ -0.9113,
19
+ 0.1075,
20
+ -0.1745,
21
+ 0.9653,
22
+ -0.1517,
23
+ 1.5508,
24
+ 0.4134,
25
+ -0.0715,
26
+ 0.5517,
27
+ -0.3632,
28
+ -0.1922,
29
+ -0.9497,
30
+ 0.2503,
31
+ -0.2921
32
+ ],
33
+ "latents_std": [
34
+ 2.8184,
35
+ 1.4541,
36
+ 2.3275,
37
+ 2.6558,
38
+ 1.2196,
39
+ 1.7708,
40
+ 2.6052,
41
+ 2.0743,
42
+ 3.2687,
43
+ 2.1526,
44
+ 2.8652,
45
+ 1.5579,
46
+ 1.6382,
47
+ 1.1253,
48
+ 2.8251,
49
+ 1.916
50
+ ],
51
+ "num_res_blocks": 2,
52
+ "temperal_downsample": [
53
+ false,
54
+ true,
55
+ true
56
+ ],
57
+ "z_dim": 16
58
+ }
bf16/vae/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab1b61103959913d6c7e628cf793dbb2ca4726a40a3b3ae206c52b8e75bf6f08
3
+ size 507591892
q4/model_index.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Krea2Pipeline",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "FlowMatchEulerDiscreteScheduler"
7
+ ],
8
+ "text_encoder": [
9
+ "transformers",
10
+ "Qwen3VLModel"
11
+ ],
12
+ "text_encoder_select_layers": [
13
+ 2,
14
+ 5,
15
+ 8,
16
+ 11,
17
+ 14,
18
+ 17,
19
+ 20,
20
+ 23,
21
+ 26,
22
+ 29,
23
+ 32,
24
+ 35
25
+ ],
26
+ "tokenizer": [
27
+ "transformers",
28
+ "Qwen2Tokenizer"
29
+ ],
30
+ "transformer": [
31
+ "diffusers",
32
+ "Krea2Transformer2DModel"
33
+ ],
34
+ "vae": [
35
+ "diffusers",
36
+ "AutoencoderKLQwenImage"
37
+ ],
38
+ "is_distilled": false,
39
+ "patch_size": 2
40
+ }
q4/scheduler/scheduler_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "FlowMatchEulerDiscreteScheduler",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "base_image_seq_len": 256,
5
+ "base_shift": 0.5,
6
+ "invert_sigmas": false,
7
+ "max_image_seq_len": 6400,
8
+ "max_shift": 1.15,
9
+ "num_train_timesteps": 1000,
10
+ "shift": 1.0,
11
+ "shift_terminal": null,
12
+ "stochastic_sampling": false,
13
+ "time_shift_type": "exponential",
14
+ "use_beta_sigmas": false,
15
+ "use_dynamic_shifting": true,
16
+ "use_exponential_sigmas": false,
17
+ "use_karras_sigmas": false
18
+ }
q4/text_encoder/config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3VLModel"
4
+ ],
5
+ "dtype": "bfloat16",
6
+ "image_token_id": 151655,
7
+ "model_type": "qwen3_vl",
8
+ "text_config": {
9
+ "attention_bias": false,
10
+ "attention_dropout": 0.0,
11
+ "bos_token_id": 151643,
12
+ "dtype": "bfloat16",
13
+ "eos_token_id": 151645,
14
+ "head_dim": 128,
15
+ "hidden_act": "silu",
16
+ "hidden_size": 2560,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 9728,
19
+ "max_position_embeddings": 262144,
20
+ "model_type": "qwen3_vl_text",
21
+ "num_attention_heads": 32,
22
+ "num_hidden_layers": 36,
23
+ "num_key_value_heads": 8,
24
+ "pad_token_id": null,
25
+ "rms_norm_eps": 1e-6,
26
+ "rope_parameters": {
27
+ "mrope_interleaved": true,
28
+ "mrope_section": [
29
+ 24,
30
+ 20,
31
+ 20
32
+ ],
33
+ "rope_theta": 5000000,
34
+ "rope_type": "default"
35
+ },
36
+ "tie_word_embeddings": true,
37
+ "use_cache": true,
38
+ "vocab_size": 151936
39
+ },
40
+ "tie_word_embeddings": true,
41
+ "transformers_version": "5.2.0",
42
+ "video_token_id": 151656,
43
+ "vision_config": {
44
+ "deepstack_visual_indexes": [
45
+ 5,
46
+ 11,
47
+ 17
48
+ ],
49
+ "depth": 24,
50
+ "dtype": "bfloat16",
51
+ "hidden_act": "gelu_pytorch_tanh",
52
+ "hidden_size": 1024,
53
+ "in_channels": 3,
54
+ "initializer_range": 0.02,
55
+ "intermediate_size": 4096,
56
+ "model_type": "qwen3_vl",
57
+ "num_heads": 16,
58
+ "num_position_embeddings": 2304,
59
+ "out_hidden_size": 2560,
60
+ "patch_size": 16,
61
+ "spatial_merge_size": 2,
62
+ "temporal_patch_size": 2
63
+ },
64
+ "vision_end_token_id": 151653,
65
+ "vision_start_token_id": 151652,
66
+ "quantization": {
67
+ "bits": 4,
68
+ "group_size": 64
69
+ }
70
+ }
q4/text_encoder/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b427dd291d2f6e6d4ba2212c32f17ea3e393be43c88cca89ea3156da43d449e
3
+ size 3652884391
q4/tokenizer/chat_template.jinja ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0].role == 'system' %}
4
+ {%- if messages[0].content is string %}
5
+ {{- messages[0].content }}
6
+ {%- else %}
7
+ {%- for content in messages[0].content %}
8
+ {%- if 'text' in content %}
9
+ {{- content.text }}
10
+ {%- endif %}
11
+ {%- endfor %}
12
+ {%- endif %}
13
+ {{- '\n\n' }}
14
+ {%- endif %}
15
+ {{- "# 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>" }}
16
+ {%- for tool in tools %}
17
+ {{- "\n" }}
18
+ {{- tool | tojson }}
19
+ {%- endfor %}
20
+ {{- "\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" }}
21
+ {%- else %}
22
+ {%- if messages[0].role == 'system' %}
23
+ {{- '<|im_start|>system\n' }}
24
+ {%- if messages[0].content is string %}
25
+ {{- messages[0].content }}
26
+ {%- else %}
27
+ {%- for content in messages[0].content %}
28
+ {%- if 'text' in content %}
29
+ {{- content.text }}
30
+ {%- endif %}
31
+ {%- endfor %}
32
+ {%- endif %}
33
+ {{- '<|im_end|>\n' }}
34
+ {%- endif %}
35
+ {%- endif %}
36
+ {%- set image_count = namespace(value=0) %}
37
+ {%- set video_count = namespace(value=0) %}
38
+ {%- for message in messages %}
39
+ {%- if message.role == "user" %}
40
+ {{- '<|im_start|>' + message.role + '\n' }}
41
+ {%- if message.content is string %}
42
+ {{- message.content }}
43
+ {%- else %}
44
+ {%- for content in message.content %}
45
+ {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
46
+ {%- set image_count.value = image_count.value + 1 %}
47
+ {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
48
+ <|vision_start|><|image_pad|><|vision_end|>
49
+ {%- elif content.type == 'video' or 'video' in content %}
50
+ {%- set video_count.value = video_count.value + 1 %}
51
+ {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
52
+ <|vision_start|><|video_pad|><|vision_end|>
53
+ {%- elif 'text' in content %}
54
+ {{- content.text }}
55
+ {%- endif %}
56
+ {%- endfor %}
57
+ {%- endif %}
58
+ {{- '<|im_end|>\n' }}
59
+ {%- elif message.role == "assistant" %}
60
+ {{- '<|im_start|>' + message.role + '\n' }}
61
+ {%- if message.content is string %}
62
+ {{- message.content }}
63
+ {%- else %}
64
+ {%- for content_item in message.content %}
65
+ {%- if 'text' in content_item %}
66
+ {{- content_item.text }}
67
+ {%- endif %}
68
+ {%- endfor %}
69
+ {%- endif %}
70
+ {%- if message.tool_calls %}
71
+ {%- for tool_call in message.tool_calls %}
72
+ {%- if (loop.first and message.content) or (not loop.first) %}
73
+ {{- '\n' }}
74
+ {%- endif %}
75
+ {%- if tool_call.function %}
76
+ {%- set tool_call = tool_call.function %}
77
+ {%- endif %}
78
+ {{- '<tool_call>\n{"name": "' }}
79
+ {{- tool_call.name }}
80
+ {{- '", "arguments": ' }}
81
+ {%- if tool_call.arguments is string %}
82
+ {{- tool_call.arguments }}
83
+ {%- else %}
84
+ {{- tool_call.arguments | tojson }}
85
+ {%- endif %}
86
+ {{- '}\n</tool_call>' }}
87
+ {%- endfor %}
88
+ {%- endif %}
89
+ {{- '<|im_end|>\n' }}
90
+ {%- elif message.role == "tool" %}
91
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
92
+ {{- '<|im_start|>user' }}
93
+ {%- endif %}
94
+ {{- '\n<tool_response>\n' }}
95
+ {%- if message.content is string %}
96
+ {{- message.content }}
97
+ {%- else %}
98
+ {%- for content in message.content %}
99
+ {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
100
+ {%- set image_count.value = image_count.value + 1 %}
101
+ {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
102
+ <|vision_start|><|image_pad|><|vision_end|>
103
+ {%- elif content.type == 'video' or 'video' in content %}
104
+ {%- set video_count.value = video_count.value + 1 %}
105
+ {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
106
+ <|vision_start|><|video_pad|><|vision_end|>
107
+ {%- elif 'text' in content %}
108
+ {{- content.text }}
109
+ {%- endif %}
110
+ {%- endfor %}
111
+ {%- endif %}
112
+ {{- '\n</tool_response>' }}
113
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
114
+ {{- '<|im_end|>\n' }}
115
+ {%- endif %}
116
+ {%- endif %}
117
+ {%- endfor %}
118
+ {%- if add_generation_prompt %}
119
+ {{- '<|im_start|>assistant\n' }}
120
+ {%- endif %}
q4/tokenizer/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be75606093db2094d7cd20f3c2f385c212750648bd6ea4fb2bf507a6a4c55506
3
+ size 11422650
q4/tokenizer/tokenizer_config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "extra_special_tokens": [
9
+ "<|im_start|>",
10
+ "<|im_end|>",
11
+ "<|object_ref_start|>",
12
+ "<|object_ref_end|>",
13
+ "<|box_start|>",
14
+ "<|box_end|>",
15
+ "<|quad_start|>",
16
+ "<|quad_end|>",
17
+ "<|vision_start|>",
18
+ "<|vision_end|>",
19
+ "<|vision_pad|>",
20
+ "<|image_pad|>",
21
+ "<|video_pad|>"
22
+ ],
23
+ "is_local": true,
24
+ "model_max_length": 262144,
25
+ "pad_token": "<|endoftext|>",
26
+ "split_special_tokens": false,
27
+ "tokenizer_class": "Qwen2Tokenizer",
28
+ "unk_token": null
29
+ }
q4/transformer/config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Krea2Transformer2DModel",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "attention_head_dim": 128,
5
+ "axes_dims_rope": [
6
+ 32,
7
+ 48,
8
+ 48
9
+ ],
10
+ "in_channels": 64,
11
+ "intermediate_size": 16384,
12
+ "norm_eps": 0.00001,
13
+ "num_attention_heads": 48,
14
+ "num_key_value_heads": 12,
15
+ "num_layers": 28,
16
+ "num_layerwise_text_blocks": 2,
17
+ "num_refiner_text_blocks": 2,
18
+ "num_text_layers": 12,
19
+ "rope_theta": 1000.0,
20
+ "text_hidden_dim": 2560,
21
+ "text_intermediate_size": 6912,
22
+ "text_num_attention_heads": 20,
23
+ "text_num_key_value_heads": 20,
24
+ "timestep_embed_dim": 256,
25
+ "quantization": {
26
+ "bits": 4,
27
+ "group_size": 64
28
+ }
29
+ }
q4/transformer/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:28e917d52f8fc8221f2754df7f108b19a270146023a3b032649c48bec4190538
3
+ size 8316800604
q4/vae/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKLQwenImage",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "_name_or_path": "Qwen/Qwen-Image",
5
+ "attn_scales": [],
6
+ "base_dim": 96,
7
+ "dim_mult": [
8
+ 1,
9
+ 2,
10
+ 4,
11
+ 4
12
+ ],
13
+ "dropout": 0.0,
14
+ "input_channels": 3,
15
+ "latents_mean": [
16
+ -0.7571,
17
+ -0.7089,
18
+ -0.9113,
19
+ 0.1075,
20
+ -0.1745,
21
+ 0.9653,
22
+ -0.1517,
23
+ 1.5508,
24
+ 0.4134,
25
+ -0.0715,
26
+ 0.5517,
27
+ -0.3632,
28
+ -0.1922,
29
+ -0.9497,
30
+ 0.2503,
31
+ -0.2921
32
+ ],
33
+ "latents_std": [
34
+ 2.8184,
35
+ 1.4541,
36
+ 2.3275,
37
+ 2.6558,
38
+ 1.2196,
39
+ 1.7708,
40
+ 2.6052,
41
+ 2.0743,
42
+ 3.2687,
43
+ 2.1526,
44
+ 2.8652,
45
+ 1.5579,
46
+ 1.6382,
47
+ 1.1253,
48
+ 2.8251,
49
+ 1.916
50
+ ],
51
+ "num_res_blocks": 2,
52
+ "temperal_downsample": [
53
+ false,
54
+ true,
55
+ true
56
+ ],
57
+ "z_dim": 16
58
+ }
q4/vae/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab1b61103959913d6c7e628cf793dbb2ca4726a40a3b3ae206c52b8e75bf6f08
3
+ size 507591892
q8/model_index.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Krea2Pipeline",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "FlowMatchEulerDiscreteScheduler"
7
+ ],
8
+ "text_encoder": [
9
+ "transformers",
10
+ "Qwen3VLModel"
11
+ ],
12
+ "text_encoder_select_layers": [
13
+ 2,
14
+ 5,
15
+ 8,
16
+ 11,
17
+ 14,
18
+ 17,
19
+ 20,
20
+ 23,
21
+ 26,
22
+ 29,
23
+ 32,
24
+ 35
25
+ ],
26
+ "tokenizer": [
27
+ "transformers",
28
+ "Qwen2Tokenizer"
29
+ ],
30
+ "transformer": [
31
+ "diffusers",
32
+ "Krea2Transformer2DModel"
33
+ ],
34
+ "vae": [
35
+ "diffusers",
36
+ "AutoencoderKLQwenImage"
37
+ ],
38
+ "is_distilled": false,
39
+ "patch_size": 2
40
+ }
q8/scheduler/scheduler_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "FlowMatchEulerDiscreteScheduler",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "base_image_seq_len": 256,
5
+ "base_shift": 0.5,
6
+ "invert_sigmas": false,
7
+ "max_image_seq_len": 6400,
8
+ "max_shift": 1.15,
9
+ "num_train_timesteps": 1000,
10
+ "shift": 1.0,
11
+ "shift_terminal": null,
12
+ "stochastic_sampling": false,
13
+ "time_shift_type": "exponential",
14
+ "use_beta_sigmas": false,
15
+ "use_dynamic_shifting": true,
16
+ "use_exponential_sigmas": false,
17
+ "use_karras_sigmas": false
18
+ }
q8/text_encoder/config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3VLModel"
4
+ ],
5
+ "dtype": "bfloat16",
6
+ "image_token_id": 151655,
7
+ "model_type": "qwen3_vl",
8
+ "text_config": {
9
+ "attention_bias": false,
10
+ "attention_dropout": 0.0,
11
+ "bos_token_id": 151643,
12
+ "dtype": "bfloat16",
13
+ "eos_token_id": 151645,
14
+ "head_dim": 128,
15
+ "hidden_act": "silu",
16
+ "hidden_size": 2560,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 9728,
19
+ "max_position_embeddings": 262144,
20
+ "model_type": "qwen3_vl_text",
21
+ "num_attention_heads": 32,
22
+ "num_hidden_layers": 36,
23
+ "num_key_value_heads": 8,
24
+ "pad_token_id": null,
25
+ "rms_norm_eps": 1e-6,
26
+ "rope_parameters": {
27
+ "mrope_interleaved": true,
28
+ "mrope_section": [
29
+ 24,
30
+ 20,
31
+ 20
32
+ ],
33
+ "rope_theta": 5000000,
34
+ "rope_type": "default"
35
+ },
36
+ "tie_word_embeddings": true,
37
+ "use_cache": true,
38
+ "vocab_size": 151936
39
+ },
40
+ "tie_word_embeddings": true,
41
+ "transformers_version": "5.2.0",
42
+ "video_token_id": 151656,
43
+ "vision_config": {
44
+ "deepstack_visual_indexes": [
45
+ 5,
46
+ 11,
47
+ 17
48
+ ],
49
+ "depth": 24,
50
+ "dtype": "bfloat16",
51
+ "hidden_act": "gelu_pytorch_tanh",
52
+ "hidden_size": 1024,
53
+ "in_channels": 3,
54
+ "initializer_range": 0.02,
55
+ "intermediate_size": 4096,
56
+ "model_type": "qwen3_vl",
57
+ "num_heads": 16,
58
+ "num_position_embeddings": 2304,
59
+ "out_hidden_size": 2560,
60
+ "patch_size": 16,
61
+ "spatial_merge_size": 2,
62
+ "temporal_patch_size": 2
63
+ },
64
+ "vision_end_token_id": 151653,
65
+ "vision_start_token_id": 151652,
66
+ "quantization": {
67
+ "bits": 8,
68
+ "group_size": 64
69
+ }
70
+ }
q8/text_encoder/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3276ccad0c7f192c2dfff8140f6ac39e57d12a452caf5e79c9815f4e2f4e87e1
3
+ size 5469542597
q8/tokenizer/chat_template.jinja ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0].role == 'system' %}
4
+ {%- if messages[0].content is string %}
5
+ {{- messages[0].content }}
6
+ {%- else %}
7
+ {%- for content in messages[0].content %}
8
+ {%- if 'text' in content %}
9
+ {{- content.text }}
10
+ {%- endif %}
11
+ {%- endfor %}
12
+ {%- endif %}
13
+ {{- '\n\n' }}
14
+ {%- endif %}
15
+ {{- "# 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>" }}
16
+ {%- for tool in tools %}
17
+ {{- "\n" }}
18
+ {{- tool | tojson }}
19
+ {%- endfor %}
20
+ {{- "\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" }}
21
+ {%- else %}
22
+ {%- if messages[0].role == 'system' %}
23
+ {{- '<|im_start|>system\n' }}
24
+ {%- if messages[0].content is string %}
25
+ {{- messages[0].content }}
26
+ {%- else %}
27
+ {%- for content in messages[0].content %}
28
+ {%- if 'text' in content %}
29
+ {{- content.text }}
30
+ {%- endif %}
31
+ {%- endfor %}
32
+ {%- endif %}
33
+ {{- '<|im_end|>\n' }}
34
+ {%- endif %}
35
+ {%- endif %}
36
+ {%- set image_count = namespace(value=0) %}
37
+ {%- set video_count = namespace(value=0) %}
38
+ {%- for message in messages %}
39
+ {%- if message.role == "user" %}
40
+ {{- '<|im_start|>' + message.role + '\n' }}
41
+ {%- if message.content is string %}
42
+ {{- message.content }}
43
+ {%- else %}
44
+ {%- for content in message.content %}
45
+ {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
46
+ {%- set image_count.value = image_count.value + 1 %}
47
+ {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
48
+ <|vision_start|><|image_pad|><|vision_end|>
49
+ {%- elif content.type == 'video' or 'video' in content %}
50
+ {%- set video_count.value = video_count.value + 1 %}
51
+ {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
52
+ <|vision_start|><|video_pad|><|vision_end|>
53
+ {%- elif 'text' in content %}
54
+ {{- content.text }}
55
+ {%- endif %}
56
+ {%- endfor %}
57
+ {%- endif %}
58
+ {{- '<|im_end|>\n' }}
59
+ {%- elif message.role == "assistant" %}
60
+ {{- '<|im_start|>' + message.role + '\n' }}
61
+ {%- if message.content is string %}
62
+ {{- message.content }}
63
+ {%- else %}
64
+ {%- for content_item in message.content %}
65
+ {%- if 'text' in content_item %}
66
+ {{- content_item.text }}
67
+ {%- endif %}
68
+ {%- endfor %}
69
+ {%- endif %}
70
+ {%- if message.tool_calls %}
71
+ {%- for tool_call in message.tool_calls %}
72
+ {%- if (loop.first and message.content) or (not loop.first) %}
73
+ {{- '\n' }}
74
+ {%- endif %}
75
+ {%- if tool_call.function %}
76
+ {%- set tool_call = tool_call.function %}
77
+ {%- endif %}
78
+ {{- '<tool_call>\n{"name": "' }}
79
+ {{- tool_call.name }}
80
+ {{- '", "arguments": ' }}
81
+ {%- if tool_call.arguments is string %}
82
+ {{- tool_call.arguments }}
83
+ {%- else %}
84
+ {{- tool_call.arguments | tojson }}
85
+ {%- endif %}
86
+ {{- '}\n</tool_call>' }}
87
+ {%- endfor %}
88
+ {%- endif %}
89
+ {{- '<|im_end|>\n' }}
90
+ {%- elif message.role == "tool" %}
91
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
92
+ {{- '<|im_start|>user' }}
93
+ {%- endif %}
94
+ {{- '\n<tool_response>\n' }}
95
+ {%- if message.content is string %}
96
+ {{- message.content }}
97
+ {%- else %}
98
+ {%- for content in message.content %}
99
+ {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
100
+ {%- set image_count.value = image_count.value + 1 %}
101
+ {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
102
+ <|vision_start|><|image_pad|><|vision_end|>
103
+ {%- elif content.type == 'video' or 'video' in content %}
104
+ {%- set video_count.value = video_count.value + 1 %}
105
+ {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
106
+ <|vision_start|><|video_pad|><|vision_end|>
107
+ {%- elif 'text' in content %}
108
+ {{- content.text }}
109
+ {%- endif %}
110
+ {%- endfor %}
111
+ {%- endif %}
112
+ {{- '\n</tool_response>' }}
113
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
114
+ {{- '<|im_end|>\n' }}
115
+ {%- endif %}
116
+ {%- endif %}
117
+ {%- endfor %}
118
+ {%- if add_generation_prompt %}
119
+ {{- '<|im_start|>assistant\n' }}
120
+ {%- endif %}
q8/tokenizer/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be75606093db2094d7cd20f3c2f385c212750648bd6ea4fb2bf507a6a4c55506
3
+ size 11422650
q8/tokenizer/tokenizer_config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "extra_special_tokens": [
9
+ "<|im_start|>",
10
+ "<|im_end|>",
11
+ "<|object_ref_start|>",
12
+ "<|object_ref_end|>",
13
+ "<|box_start|>",
14
+ "<|box_end|>",
15
+ "<|quad_start|>",
16
+ "<|quad_end|>",
17
+ "<|vision_start|>",
18
+ "<|vision_end|>",
19
+ "<|vision_pad|>",
20
+ "<|image_pad|>",
21
+ "<|video_pad|>"
22
+ ],
23
+ "is_local": true,
24
+ "model_max_length": 262144,
25
+ "pad_token": "<|endoftext|>",
26
+ "split_special_tokens": false,
27
+ "tokenizer_class": "Qwen2Tokenizer",
28
+ "unk_token": null
29
+ }
q8/transformer/config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Krea2Transformer2DModel",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "attention_head_dim": 128,
5
+ "axes_dims_rope": [
6
+ 32,
7
+ 48,
8
+ 48
9
+ ],
10
+ "in_channels": 64,
11
+ "intermediate_size": 16384,
12
+ "norm_eps": 0.00001,
13
+ "num_attention_heads": 48,
14
+ "num_key_value_heads": 12,
15
+ "num_layers": 28,
16
+ "num_layerwise_text_blocks": 2,
17
+ "num_refiner_text_blocks": 2,
18
+ "num_text_layers": 12,
19
+ "rope_theta": 1000.0,
20
+ "text_hidden_dim": 2560,
21
+ "text_intermediate_size": 6912,
22
+ "text_num_attention_heads": 20,
23
+ "text_num_key_value_heads": 20,
24
+ "timestep_embed_dim": 256,
25
+ "quantization": {
26
+ "bits": 8,
27
+ "group_size": 64
28
+ }
29
+ }
q8/transformer/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e55cfb87e18d9d1d54325ed4ed2ab623e063bf420785e3ff05496e8616309435
3
+ size 14566052205
q8/vae/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKLQwenImage",
3
+ "_diffusers_version": "0.39.0.dev0",
4
+ "_name_or_path": "Qwen/Qwen-Image",
5
+ "attn_scales": [],
6
+ "base_dim": 96,
7
+ "dim_mult": [
8
+ 1,
9
+ 2,
10
+ 4,
11
+ 4
12
+ ],
13
+ "dropout": 0.0,
14
+ "input_channels": 3,
15
+ "latents_mean": [
16
+ -0.7571,
17
+ -0.7089,
18
+ -0.9113,
19
+ 0.1075,
20
+ -0.1745,
21
+ 0.9653,
22
+ -0.1517,
23
+ 1.5508,
24
+ 0.4134,
25
+ -0.0715,
26
+ 0.5517,
27
+ -0.3632,
28
+ -0.1922,
29
+ -0.9497,
30
+ 0.2503,
31
+ -0.2921
32
+ ],
33
+ "latents_std": [
34
+ 2.8184,
35
+ 1.4541,
36
+ 2.3275,
37
+ 2.6558,
38
+ 1.2196,
39
+ 1.7708,
40
+ 2.6052,
41
+ 2.0743,
42
+ 3.2687,
43
+ 2.1526,
44
+ 2.8652,
45
+ 1.5579,
46
+ 1.6382,
47
+ 1.1253,
48
+ 2.8251,
49
+ 1.916
50
+ ],
51
+ "num_res_blocks": 2,
52
+ "temperal_downsample": [
53
+ false,
54
+ true,
55
+ true
56
+ ],
57
+ "z_dim": 16
58
+ }
q8/vae/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab1b61103959913d6c7e628cf793dbb2ca4726a40a3b3ae206c52b8e75bf6f08
3
+ size 507591892
upload.log ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ It seems you are trying to upload a large folder at once. This might take some time and then fail if the folder is too large. For such cases, it is recommended to upload in smaller batches or to use `HfApi().upload_large_folder(...)`/`hf upload-large-folder` instead. For more details, check out https://huggingface.co/docs/huggingface_hub/main/en/guides/upload#upload-a-large-folder.
2
+ Start hashing 39 files.
3
+ Finished hashing 39 files.