diff --git a/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00001-of-00003.safetensors b/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00001-of-00003.safetensors deleted file mode 100644 index 2ae58d465c44801bc5281dde50aab058d9ea739b..0000000000000000000000000000000000000000 --- a/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00001-of-00003.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a1842c19379893f0b33005372b6b16652d9ff22ecc9b411799e037e83ea71206 -size 9973578592 diff --git a/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00002-of-00003.safetensors b/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00002-of-00003.safetensors deleted file mode 100644 index 5425b9e84fb85b503f64249a01dc5611d7a91dcd..0000000000000000000000000000000000000000 --- a/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00002-of-00003.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:3290d8f63d8cba6d1104533f8c09ab82b4ec9042f65c9320d7f47a56d5253ce8 -size 9987326072 diff --git a/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00003-of-00003.safetensors b/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00003-of-00003.safetensors deleted file mode 100644 index 4617f0021577fbc54cc744090cf3f5399219a5a6..0000000000000000000000000000000000000000 --- a/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model-00003-of-00003.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f526e821117858f9ac08645cc1e1905a6bca1c09cae5c09d7f438b88d92a5f8f -size 4588131352 diff --git a/Qwen-Image-12B-40steps/model_index.json b/Qwen-Image-12B-8steps/model_index.json similarity index 81% rename from Qwen-Image-12B-40steps/model_index.json rename to Qwen-Image-12B-8steps/model_index.json index 42906a5375fb88345656e3938170dd62c2b1bd46..19b9f297f1d807c3085f08271d37070c3ede9ee5 100644 --- a/Qwen-Image-12B-40steps/model_index.json +++ b/Qwen-Image-12B-8steps/model_index.json @@ -1,7 +1,7 @@ { "_class_name": "QwenImagePipeline", "_diffusers_version": "0.36.0.dev0", - "_name_or_path": "/mnt/workspace/group/models/Qwen-Image/Qwen-Image", + "_name_or_path": "/mnt/workspace/group/models/Qwen-Image", "scheduler": [ "diffusers", "FlowMatchEulerDiscreteScheduler" @@ -15,7 +15,7 @@ "Qwen2Tokenizer" ], "transformer": [ - "diffusers", + "transformer_qwenimage_pruning", "QwenImageTransformer2DModel" ], "vae": [ diff --git a/Qwen-Image-13B/scheduler/scheduler_config.json b/Qwen-Image-12B-8steps/scheduler/scheduler_config.json similarity index 100% rename from Qwen-Image-13B/scheduler/scheduler_config.json rename to Qwen-Image-12B-8steps/scheduler/scheduler_config.json diff --git a/Qwen-Image-13B/text_encoder/config.json b/Qwen-Image-12B-8steps/text_encoder/config.json similarity index 96% rename from Qwen-Image-13B/text_encoder/config.json rename to Qwen-Image-12B-8steps/text_encoder/config.json index 3c8abaa7c93408b12547fc0589e30c117ea2b26a..1cafff30eef28323f9b71da12041ddda24637312 100644 --- a/Qwen-Image-13B/text_encoder/config.json +++ b/Qwen-Image-12B-8steps/text_encoder/config.json @@ -4,7 +4,6 @@ ], "attention_dropout": 0.0, "bos_token_id": 151643, - "dtype": "bfloat16", "eos_token_id": 151645, "hidden_act": "silu", "hidden_size": 3584, @@ -35,7 +34,6 @@ ], "attention_dropout": 0.0, "bos_token_id": 151643, - "dtype": "bfloat16", "eos_token_id": 151645, "hidden_act": "silu", "hidden_size": 3584, @@ -90,6 +88,7 @@ }, "rope_theta": 1000000.0, "sliding_window": null, + "torch_dtype": "bfloat16", "use_cache": true, "use_sliding_window": false, "video_token_id": null, @@ -99,13 +98,13 @@ "vocab_size": 152064 }, "tie_word_embeddings": false, - "transformers_version": "4.56.1", + "torch_dtype": "bfloat16", + "transformers_version": "4.54.1", "use_cache": true, "use_sliding_window": false, "video_token_id": 151656, "vision_config": { "depth": 32, - "dtype": "bfloat16", "fullatt_block_indexes": [ 7, 15, @@ -126,6 +125,7 @@ "spatial_patch_size": 14, "temporal_patch_size": 2, "tokens_per_second": 2, + "torch_dtype": "bfloat16", "window_size": 112 }, "vision_end_token_id": 151653, diff --git a/Qwen-Image-13B/text_encoder/generation_config.json b/Qwen-Image-12B-8steps/text_encoder/generation_config.json similarity index 85% rename from Qwen-Image-13B/text_encoder/generation_config.json rename to Qwen-Image-12B-8steps/text_encoder/generation_config.json index 32704b093110a4e700576ffad082018002bd8bc7..4c806852785092f055baeeeb1828424a8394adf3 100644 --- a/Qwen-Image-13B/text_encoder/generation_config.json +++ b/Qwen-Image-12B-8steps/text_encoder/generation_config.json @@ -10,5 +10,5 @@ "temperature": 0.1, "top_k": 1, "top_p": 0.001, - "transformers_version": "4.56.1" + "transformers_version": "4.54.1" } diff --git a/Qwen-Image-12B-40steps/text_encoder/model-00001-of-00004.safetensors b/Qwen-Image-12B-8steps/text_encoder/model-00001-of-00004.safetensors similarity index 100% rename from Qwen-Image-12B-40steps/text_encoder/model-00001-of-00004.safetensors rename to Qwen-Image-12B-8steps/text_encoder/model-00001-of-00004.safetensors diff --git a/Qwen-Image-12B-40steps/text_encoder/model-00002-of-00004.safetensors b/Qwen-Image-12B-8steps/text_encoder/model-00002-of-00004.safetensors similarity index 100% rename from Qwen-Image-12B-40steps/text_encoder/model-00002-of-00004.safetensors rename to Qwen-Image-12B-8steps/text_encoder/model-00002-of-00004.safetensors diff --git a/Qwen-Image-12B-40steps/text_encoder/model-00003-of-00004.safetensors b/Qwen-Image-12B-8steps/text_encoder/model-00003-of-00004.safetensors similarity index 100% rename from Qwen-Image-12B-40steps/text_encoder/model-00003-of-00004.safetensors rename to Qwen-Image-12B-8steps/text_encoder/model-00003-of-00004.safetensors diff --git a/Qwen-Image-12B-40steps/text_encoder/model-00004-of-00004.safetensors b/Qwen-Image-12B-8steps/text_encoder/model-00004-of-00004.safetensors similarity index 100% rename from Qwen-Image-12B-40steps/text_encoder/model-00004-of-00004.safetensors rename to Qwen-Image-12B-8steps/text_encoder/model-00004-of-00004.safetensors diff --git a/Qwen-Image-12B-40steps/text_encoder/model.safetensors.index.json b/Qwen-Image-12B-8steps/text_encoder/model.safetensors.index.json similarity index 100% rename from Qwen-Image-12B-40steps/text_encoder/model.safetensors.index.json rename to Qwen-Image-12B-8steps/text_encoder/model.safetensors.index.json diff --git a/Qwen-Image-12B-40steps/tokenizer/added_tokens.json b/Qwen-Image-12B-8steps/tokenizer/added_tokens.json similarity index 100% rename from Qwen-Image-12B-40steps/tokenizer/added_tokens.json rename to Qwen-Image-12B-8steps/tokenizer/added_tokens.json diff --git a/Qwen-Image-12B-40steps/tokenizer/chat_template.jinja b/Qwen-Image-12B-8steps/tokenizer/chat_template.jinja similarity index 100% rename from Qwen-Image-12B-40steps/tokenizer/chat_template.jinja rename to Qwen-Image-12B-8steps/tokenizer/chat_template.jinja diff --git a/Qwen-Image-12B-40steps/tokenizer/merges.txt b/Qwen-Image-12B-8steps/tokenizer/merges.txt similarity index 100% rename from Qwen-Image-12B-40steps/tokenizer/merges.txt rename to Qwen-Image-12B-8steps/tokenizer/merges.txt diff --git a/Qwen-Image-12B-40steps/tokenizer/special_tokens_map.json b/Qwen-Image-12B-8steps/tokenizer/special_tokens_map.json similarity index 100% rename from Qwen-Image-12B-40steps/tokenizer/special_tokens_map.json rename to Qwen-Image-12B-8steps/tokenizer/special_tokens_map.json diff --git a/Qwen-Image-12B-40steps/tokenizer/tokenizer_config.json b/Qwen-Image-12B-8steps/tokenizer/tokenizer_config.json similarity index 100% rename from Qwen-Image-12B-40steps/tokenizer/tokenizer_config.json rename to Qwen-Image-12B-8steps/tokenizer/tokenizer_config.json diff --git a/Qwen-Image-12B-40steps/tokenizer/vocab.json b/Qwen-Image-12B-8steps/tokenizer/vocab.json similarity index 100% rename from Qwen-Image-12B-40steps/tokenizer/vocab.json rename to Qwen-Image-12B-8steps/tokenizer/vocab.json diff --git a/Qwen-Image-12B-40steps/transformer/config.json b/Qwen-Image-12B-8steps/transformer/config.json similarity index 66% rename from Qwen-Image-12B-40steps/transformer/config.json rename to Qwen-Image-12B-8steps/transformer/config.json index d47757c1aad96536cf7ba1d61bbb41b0f8f6602b..460d9b047ee5472be21f23c62b14eff9378db804 100644 --- a/Qwen-Image-12B-40steps/transformer/config.json +++ b/Qwen-Image-12B-8steps/transformer/config.json @@ -11,7 +11,19 @@ "in_channels": 64, "joint_attention_dim": 3584, "num_attention_heads": 24, - "num_layers": 36, + "num_layers": 41, "out_channels": 16, - "patch_size": 2 + "patch_size": 2, + "single_stream_index": [ + 18, + 27, + 29, + 32, + 35, + 36, + 37, + 38, + 39, + 40 + ] } diff --git a/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00001-of-00003.safetensors b/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00001-of-00003.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..378d24d190fd9ecdcc1017f0e2acd862e47be596 --- /dev/null +++ b/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00001-of-00003.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1593c9c51f38410eda7732eff281b642ee55d97b32acd04e15ca01b492b646f1 +size 9973578592 diff --git a/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00002-of-00003.safetensors b/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00002-of-00003.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..ba4541647146f6824e02d91a93e4b1e133fdbcc4 --- /dev/null +++ b/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00002-of-00003.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7f63875668e3676f606e4bd60547ade76e02ad88f48d13673d58dcb08a479caa +size 9987349288 diff --git a/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00003-of-00003.safetensors b/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00003-of-00003.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..04f642fba486a0a098cf22d990d34a5261e57f75 --- /dev/null +++ b/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model-00003-of-00003.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a0325e84cb03b7439412378075c21bdd0c002e1b4e18c82633024f39e474412f +size 5532152808 diff --git a/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model.safetensors.index.json b/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model.safetensors.index.json similarity index 89% rename from Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model.safetensors.index.json rename to Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model.safetensors.index.json index 145d9baf79e817558c36c8f434803dd30cb9be63..131970f6a4d7dd0a5056121ad88e6598d858dfec 100644 --- a/Qwen-Image-12B-40steps/transformer/diffusion_pytorch_model.safetensors.index.json +++ b/Qwen-Image-12B-8steps/transformer/diffusion_pytorch_model.safetensors.index.json @@ -1,6 +1,6 @@ { "metadata": { - "total_size": 24548899968 + "total_size": 25492930688 }, "weight_map": { "img_in.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", @@ -359,12 +359,8 @@ "transformer_blocks.18.img_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.18.img_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.18.img_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.18.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.18.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.18.txt_mlp.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.18.txt_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.18.txt_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.18.txt_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.18.txt_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.18.txt_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.19.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.19.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.19.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", @@ -679,12 +675,8 @@ "transformer_blocks.27.img_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.27.img_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.27.img_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.27.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.27.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.27.txt_mlp.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.27.txt_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.27.txt_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.27.txt_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.27.txt_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.27.txt_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.28.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.28.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.28.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", @@ -719,36 +711,32 @@ "transformer_blocks.28.txt_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.img_mlp.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.img_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.img_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.img_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.txt_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.txt_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.txt_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.29.txt_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.txt_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.txt_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.3.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.3.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", @@ -781,32 +769,32 @@ "transformer_blocks.3.txt_mlp.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.3.txt_mod.1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.3.txt_mod.1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", - "transformer_blocks.30.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.30.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.img_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.img_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.30.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.txt_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", @@ -871,12 +859,8 @@ "transformer_blocks.32.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.32.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.32.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.32.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.32.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.32.txt_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.32.txt_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.32.txt_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.32.txt_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.32.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.32.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.33.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.33.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.33.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", @@ -967,12 +951,120 @@ "transformer_blocks.35.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.35.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.35.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.35.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.35.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.35.txt_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.35.txt_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.35.txt_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.35.txt_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.35.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.35.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.36.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.37.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.38.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.39.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.4.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.4.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", @@ -1005,6 +1097,34 @@ "transformer_blocks.4.txt_mlp.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.4.txt_mod.1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.4.txt_mod.1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", + "transformer_blocks.40.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.40.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.5.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.5.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", diff --git a/Qwen-Image-12B-40steps/vae/config.json b/Qwen-Image-12B-8steps/vae/config.json similarity index 90% rename from Qwen-Image-12B-40steps/vae/config.json rename to Qwen-Image-12B-8steps/vae/config.json index 9a9b6663e7eacc5a8265095ca939314379afc338..f60c475fa6438d6095978157e725d6d16441cc3a 100644 --- a/Qwen-Image-12B-40steps/vae/config.json +++ b/Qwen-Image-12B-8steps/vae/config.json @@ -1,7 +1,7 @@ { "_class_name": "AutoencoderKLQwenImage", "_diffusers_version": "0.36.0.dev0", - "_name_or_path": "/mnt/workspace/group/models/Qwen-Image/Qwen-Image/vae", + "_name_or_path": "/mnt/workspace/group/models/Qwen-Image/vae", "attn_scales": [], "base_dim": 96, "dim_mult": [ diff --git a/Qwen-Image-12B-40steps/vae/diffusion_pytorch_model.safetensors b/Qwen-Image-12B-8steps/vae/diffusion_pytorch_model.safetensors similarity index 100% rename from Qwen-Image-12B-40steps/vae/diffusion_pytorch_model.safetensors rename to Qwen-Image-12B-8steps/vae/diffusion_pytorch_model.safetensors diff --git a/Qwen-Image-12B/model_index.json b/Qwen-Image-12B/model_index.json index 19b9f297f1d807c3085f08271d37070c3ede9ee5..42906a5375fb88345656e3938170dd62c2b1bd46 100644 --- a/Qwen-Image-12B/model_index.json +++ b/Qwen-Image-12B/model_index.json @@ -1,7 +1,7 @@ { "_class_name": "QwenImagePipeline", "_diffusers_version": "0.36.0.dev0", - "_name_or_path": "/mnt/workspace/group/models/Qwen-Image", + "_name_or_path": "/mnt/workspace/group/models/Qwen-Image/Qwen-Image", "scheduler": [ "diffusers", "FlowMatchEulerDiscreteScheduler" @@ -15,7 +15,7 @@ "Qwen2Tokenizer" ], "transformer": [ - "transformer_qwenimage_pruning", + "diffusers", "QwenImageTransformer2DModel" ], "vae": [ diff --git a/Qwen-Image-12B/scheduler/scheduler_config.json b/Qwen-Image-12B/scheduler/scheduler_config.json index e595b3fb938bf9354bd01f3635b90b83b2330233..f400c338630cdaa261c6575648f374eba547b9c6 100644 --- a/Qwen-Image-12B/scheduler/scheduler_config.json +++ b/Qwen-Image-12B/scheduler/scheduler_config.json @@ -2,13 +2,13 @@ "_class_name": "FlowMatchEulerDiscreteScheduler", "_diffusers_version": "0.36.0.dev0", "base_image_seq_len": 256, - "base_shift": 1.0986122886681098, + "base_shift": 0.5, "invert_sigmas": false, "max_image_seq_len": 8192, - "max_shift": 1.0986122886681098, + "max_shift": 0.9, "num_train_timesteps": 1000, "shift": 1.0, - "shift_terminal": null, + "shift_terminal": 0.02, "stochastic_sampling": false, "time_shift_type": "exponential", "use_beta_sigmas": false, diff --git a/Qwen-Image-12B/text_encoder/config.json b/Qwen-Image-12B/text_encoder/config.json index 1cafff30eef28323f9b71da12041ddda24637312..3c8abaa7c93408b12547fc0589e30c117ea2b26a 100644 --- a/Qwen-Image-12B/text_encoder/config.json +++ b/Qwen-Image-12B/text_encoder/config.json @@ -4,6 +4,7 @@ ], "attention_dropout": 0.0, "bos_token_id": 151643, + "dtype": "bfloat16", "eos_token_id": 151645, "hidden_act": "silu", "hidden_size": 3584, @@ -34,6 +35,7 @@ ], "attention_dropout": 0.0, "bos_token_id": 151643, + "dtype": "bfloat16", "eos_token_id": 151645, "hidden_act": "silu", "hidden_size": 3584, @@ -88,7 +90,6 @@ }, "rope_theta": 1000000.0, "sliding_window": null, - "torch_dtype": "bfloat16", "use_cache": true, "use_sliding_window": false, "video_token_id": null, @@ -98,13 +99,13 @@ "vocab_size": 152064 }, "tie_word_embeddings": false, - "torch_dtype": "bfloat16", - "transformers_version": "4.54.1", + "transformers_version": "4.56.1", "use_cache": true, "use_sliding_window": false, "video_token_id": 151656, "vision_config": { "depth": 32, + "dtype": "bfloat16", "fullatt_block_indexes": [ 7, 15, @@ -125,7 +126,6 @@ "spatial_patch_size": 14, "temporal_patch_size": 2, "tokens_per_second": 2, - "torch_dtype": "bfloat16", "window_size": 112 }, "vision_end_token_id": 151653, diff --git a/Qwen-Image-12B/text_encoder/generation_config.json b/Qwen-Image-12B/text_encoder/generation_config.json index 4c806852785092f055baeeeb1828424a8394adf3..32704b093110a4e700576ffad082018002bd8bc7 100644 --- a/Qwen-Image-12B/text_encoder/generation_config.json +++ b/Qwen-Image-12B/text_encoder/generation_config.json @@ -10,5 +10,5 @@ "temperature": 0.1, "top_k": 1, "top_p": 0.001, - "transformers_version": "4.54.1" + "transformers_version": "4.56.1" } diff --git a/Qwen-Image-12B/transformer/config.json b/Qwen-Image-12B/transformer/config.json index 460d9b047ee5472be21f23c62b14eff9378db804..d47757c1aad96536cf7ba1d61bbb41b0f8f6602b 100644 --- a/Qwen-Image-12B/transformer/config.json +++ b/Qwen-Image-12B/transformer/config.json @@ -11,19 +11,7 @@ "in_channels": 64, "joint_attention_dim": 3584, "num_attention_heads": 24, - "num_layers": 41, + "num_layers": 36, "out_channels": 16, - "patch_size": 2, - "single_stream_index": [ - 18, - 27, - 29, - 32, - 35, - 36, - 37, - 38, - 39, - 40 - ] + "patch_size": 2 } diff --git a/Qwen-Image-12B/transformer/diffusion_pytorch_model-00001-of-00003.safetensors b/Qwen-Image-12B/transformer/diffusion_pytorch_model-00001-of-00003.safetensors index 378d24d190fd9ecdcc1017f0e2acd862e47be596..2ae58d465c44801bc5281dde50aab058d9ea739b 100644 --- a/Qwen-Image-12B/transformer/diffusion_pytorch_model-00001-of-00003.safetensors +++ b/Qwen-Image-12B/transformer/diffusion_pytorch_model-00001-of-00003.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:1593c9c51f38410eda7732eff281b642ee55d97b32acd04e15ca01b492b646f1 +oid sha256:a1842c19379893f0b33005372b6b16652d9ff22ecc9b411799e037e83ea71206 size 9973578592 diff --git a/Qwen-Image-12B/transformer/diffusion_pytorch_model-00002-of-00003.safetensors b/Qwen-Image-12B/transformer/diffusion_pytorch_model-00002-of-00003.safetensors index ba4541647146f6824e02d91a93e4b1e133fdbcc4..5425b9e84fb85b503f64249a01dc5611d7a91dcd 100644 --- a/Qwen-Image-12B/transformer/diffusion_pytorch_model-00002-of-00003.safetensors +++ b/Qwen-Image-12B/transformer/diffusion_pytorch_model-00002-of-00003.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:7f63875668e3676f606e4bd60547ade76e02ad88f48d13673d58dcb08a479caa -size 9987349288 +oid sha256:3290d8f63d8cba6d1104533f8c09ab82b4ec9042f65c9320d7f47a56d5253ce8 +size 9987326072 diff --git a/Qwen-Image-12B/transformer/diffusion_pytorch_model-00003-of-00003.safetensors b/Qwen-Image-12B/transformer/diffusion_pytorch_model-00003-of-00003.safetensors index 04f642fba486a0a098cf22d990d34a5261e57f75..4617f0021577fbc54cc744090cf3f5399219a5a6 100644 --- a/Qwen-Image-12B/transformer/diffusion_pytorch_model-00003-of-00003.safetensors +++ b/Qwen-Image-12B/transformer/diffusion_pytorch_model-00003-of-00003.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:a0325e84cb03b7439412378075c21bdd0c002e1b4e18c82633024f39e474412f -size 5532152808 +oid sha256:f526e821117858f9ac08645cc1e1905a6bca1c09cae5c09d7f438b88d92a5f8f +size 4588131352 diff --git a/Qwen-Image-12B/transformer/diffusion_pytorch_model.safetensors.index.json b/Qwen-Image-12B/transformer/diffusion_pytorch_model.safetensors.index.json index 131970f6a4d7dd0a5056121ad88e6598d858dfec..145d9baf79e817558c36c8f434803dd30cb9be63 100644 --- a/Qwen-Image-12B/transformer/diffusion_pytorch_model.safetensors.index.json +++ b/Qwen-Image-12B/transformer/diffusion_pytorch_model.safetensors.index.json @@ -1,6 +1,6 @@ { "metadata": { - "total_size": 25492930688 + "total_size": 24548899968 }, "weight_map": { "img_in.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", @@ -359,8 +359,12 @@ "transformer_blocks.18.img_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.18.img_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.18.img_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.18.txt_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.18.txt_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.18.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.18.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.18.txt_mlp.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.18.txt_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.18.txt_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.18.txt_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.19.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.19.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.19.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", @@ -675,8 +679,12 @@ "transformer_blocks.27.img_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.27.img_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.27.img_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.27.txt_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.27.txt_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.27.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.27.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.27.txt_mlp.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.27.txt_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.27.txt_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.27.txt_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.28.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.28.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.28.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", @@ -711,32 +719,36 @@ "transformer_blocks.28.txt_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.29.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.29.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.29.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.29.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.img_mlp.net.2.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.img_mlp.net.2.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.29.img_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", "transformer_blocks.29.img_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.txt_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.29.txt_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.29.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.txt_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.txt_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.txt_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.29.txt_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.3.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.3.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.3.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", @@ -769,32 +781,32 @@ "transformer_blocks.3.txt_mlp.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.3.txt_mod.1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.3.txt_mod.1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", - "transformer_blocks.30.attn.add_k_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.add_q_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.add_q_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.add_v_proj.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_add_out.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_k.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_out.0.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_out.0.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_q.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_v.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.30.img_mod.1.bias": "diffusion_pytorch_model-00002-of-00003.safetensors", - "transformer_blocks.30.img_mod.1.weight": "diffusion_pytorch_model-00002-of-00003.safetensors", + "transformer_blocks.30.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.30.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.30.txt_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", @@ -859,8 +871,12 @@ "transformer_blocks.32.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.32.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.32.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.32.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.32.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.32.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.32.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.32.txt_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.32.txt_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.32.txt_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.32.txt_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.33.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.33.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.33.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", @@ -951,120 +967,12 @@ "transformer_blocks.35.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.35.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.35.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.35.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.35.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.36.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.37.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.38.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.39.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.35.txt_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.35.txt_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.35.txt_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.35.txt_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.35.txt_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", + "transformer_blocks.35.txt_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.4.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.4.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.4.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", @@ -1097,34 +1005,6 @@ "transformer_blocks.4.txt_mlp.net.2.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.4.txt_mod.1.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.4.txt_mod.1.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", - "transformer_blocks.40.attn.add_k_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.add_k_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.add_q_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.add_q_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.add_v_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.add_v_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.norm_added_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.norm_added_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.norm_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.norm_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_add_out.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_add_out.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_k.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_k.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_out.0.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_out.0.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_q.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_q.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_v.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.attn.to_v.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.img_mlp.net.0.proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.img_mlp.net.0.proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.img_mlp.net.2.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.img_mlp.net.2.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.img_mod.1.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.img_mod.1.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.txt_proj.bias": "diffusion_pytorch_model-00003-of-00003.safetensors", - "transformer_blocks.40.txt_proj.weight": "diffusion_pytorch_model-00003-of-00003.safetensors", "transformer_blocks.5.attn.add_k_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.5.attn.add_k_proj.weight": "diffusion_pytorch_model-00001-of-00003.safetensors", "transformer_blocks.5.attn.add_q_proj.bias": "diffusion_pytorch_model-00001-of-00003.safetensors", diff --git a/Qwen-Image-12B/vae/config.json b/Qwen-Image-12B/vae/config.json index f60c475fa6438d6095978157e725d6d16441cc3a..9a9b6663e7eacc5a8265095ca939314379afc338 100644 --- a/Qwen-Image-12B/vae/config.json +++ b/Qwen-Image-12B/vae/config.json @@ -1,7 +1,7 @@ { "_class_name": "AutoencoderKLQwenImage", "_diffusers_version": "0.36.0.dev0", - "_name_or_path": "/mnt/workspace/group/models/Qwen-Image/vae", + "_name_or_path": "/mnt/workspace/group/models/Qwen-Image/Qwen-Image/vae", "attn_scales": [], "base_dim": 96, "dim_mult": [ diff --git a/Qwen-Image-13B/model_index.json b/Qwen-Image-13B-8steps/model_index.json similarity index 100% rename from Qwen-Image-13B/model_index.json rename to Qwen-Image-13B-8steps/model_index.json diff --git a/Qwen-Image-12B-40steps/scheduler/scheduler_config.json b/Qwen-Image-13B-8steps/scheduler/scheduler_config.json similarity index 81% rename from Qwen-Image-12B-40steps/scheduler/scheduler_config.json rename to Qwen-Image-13B-8steps/scheduler/scheduler_config.json index f400c338630cdaa261c6575648f374eba547b9c6..e595b3fb938bf9354bd01f3635b90b83b2330233 100644 --- a/Qwen-Image-12B-40steps/scheduler/scheduler_config.json +++ b/Qwen-Image-13B-8steps/scheduler/scheduler_config.json @@ -2,13 +2,13 @@ "_class_name": "FlowMatchEulerDiscreteScheduler", "_diffusers_version": "0.36.0.dev0", "base_image_seq_len": 256, - "base_shift": 0.5, + "base_shift": 1.0986122886681098, "invert_sigmas": false, "max_image_seq_len": 8192, - "max_shift": 0.9, + "max_shift": 1.0986122886681098, "num_train_timesteps": 1000, "shift": 1.0, - "shift_terminal": 0.02, + "shift_terminal": null, "stochastic_sampling": false, "time_shift_type": "exponential", "use_beta_sigmas": false, diff --git a/Qwen-Image-12B-40steps/text_encoder/config.json b/Qwen-Image-13B-8steps/text_encoder/config.json similarity index 100% rename from Qwen-Image-12B-40steps/text_encoder/config.json rename to Qwen-Image-13B-8steps/text_encoder/config.json diff --git a/Qwen-Image-12B-40steps/text_encoder/generation_config.json b/Qwen-Image-13B-8steps/text_encoder/generation_config.json similarity index 100% rename from Qwen-Image-12B-40steps/text_encoder/generation_config.json rename to Qwen-Image-13B-8steps/text_encoder/generation_config.json diff --git a/Qwen-Image-13B/text_encoder/model-00001-of-00004.safetensors b/Qwen-Image-13B-8steps/text_encoder/model-00001-of-00004.safetensors similarity index 100% rename from Qwen-Image-13B/text_encoder/model-00001-of-00004.safetensors rename to Qwen-Image-13B-8steps/text_encoder/model-00001-of-00004.safetensors diff --git a/Qwen-Image-13B/text_encoder/model-00002-of-00004.safetensors b/Qwen-Image-13B-8steps/text_encoder/model-00002-of-00004.safetensors similarity index 100% rename from Qwen-Image-13B/text_encoder/model-00002-of-00004.safetensors rename to Qwen-Image-13B-8steps/text_encoder/model-00002-of-00004.safetensors diff --git a/Qwen-Image-13B/text_encoder/model-00003-of-00004.safetensors b/Qwen-Image-13B-8steps/text_encoder/model-00003-of-00004.safetensors similarity index 100% rename from Qwen-Image-13B/text_encoder/model-00003-of-00004.safetensors rename to Qwen-Image-13B-8steps/text_encoder/model-00003-of-00004.safetensors diff --git a/Qwen-Image-13B/text_encoder/model-00004-of-00004.safetensors b/Qwen-Image-13B-8steps/text_encoder/model-00004-of-00004.safetensors similarity index 100% rename from Qwen-Image-13B/text_encoder/model-00004-of-00004.safetensors rename to Qwen-Image-13B-8steps/text_encoder/model-00004-of-00004.safetensors diff --git a/Qwen-Image-13B/text_encoder/model.safetensors.index.json b/Qwen-Image-13B-8steps/text_encoder/model.safetensors.index.json similarity index 100% rename from Qwen-Image-13B/text_encoder/model.safetensors.index.json rename to Qwen-Image-13B-8steps/text_encoder/model.safetensors.index.json diff --git a/Qwen-Image-13B/tokenizer/added_tokens.json b/Qwen-Image-13B-8steps/tokenizer/added_tokens.json similarity index 100% rename from Qwen-Image-13B/tokenizer/added_tokens.json rename to Qwen-Image-13B-8steps/tokenizer/added_tokens.json diff --git a/Qwen-Image-13B/tokenizer/chat_template.jinja b/Qwen-Image-13B-8steps/tokenizer/chat_template.jinja similarity index 100% rename from Qwen-Image-13B/tokenizer/chat_template.jinja rename to Qwen-Image-13B-8steps/tokenizer/chat_template.jinja diff --git a/Qwen-Image-13B/tokenizer/merges.txt b/Qwen-Image-13B-8steps/tokenizer/merges.txt similarity index 100% rename from Qwen-Image-13B/tokenizer/merges.txt rename to Qwen-Image-13B-8steps/tokenizer/merges.txt diff --git a/Qwen-Image-13B/tokenizer/special_tokens_map.json b/Qwen-Image-13B-8steps/tokenizer/special_tokens_map.json similarity index 100% rename from Qwen-Image-13B/tokenizer/special_tokens_map.json rename to Qwen-Image-13B-8steps/tokenizer/special_tokens_map.json diff --git a/Qwen-Image-13B/tokenizer/tokenizer_config.json b/Qwen-Image-13B-8steps/tokenizer/tokenizer_config.json similarity index 100% rename from Qwen-Image-13B/tokenizer/tokenizer_config.json rename to Qwen-Image-13B-8steps/tokenizer/tokenizer_config.json diff --git a/Qwen-Image-13B/tokenizer/vocab.json b/Qwen-Image-13B-8steps/tokenizer/vocab.json similarity index 100% rename from Qwen-Image-13B/tokenizer/vocab.json rename to Qwen-Image-13B-8steps/tokenizer/vocab.json diff --git a/Qwen-Image-13B/transformer/config.json b/Qwen-Image-13B-8steps/transformer/config.json similarity index 100% rename from Qwen-Image-13B/transformer/config.json rename to Qwen-Image-13B-8steps/transformer/config.json diff --git a/Qwen-Image-13B/transformer/diffusion_pytorch_model-00001-of-00003.safetensors b/Qwen-Image-13B-8steps/transformer/diffusion_pytorch_model-00001-of-00003.safetensors similarity index 100% rename from Qwen-Image-13B/transformer/diffusion_pytorch_model-00001-of-00003.safetensors rename to Qwen-Image-13B-8steps/transformer/diffusion_pytorch_model-00001-of-00003.safetensors diff --git a/Qwen-Image-13B/transformer/diffusion_pytorch_model-00002-of-00003.safetensors b/Qwen-Image-13B-8steps/transformer/diffusion_pytorch_model-00002-of-00003.safetensors similarity index 100% rename from Qwen-Image-13B/transformer/diffusion_pytorch_model-00002-of-00003.safetensors rename to Qwen-Image-13B-8steps/transformer/diffusion_pytorch_model-00002-of-00003.safetensors diff --git a/Qwen-Image-13B/transformer/diffusion_pytorch_model-00003-of-00003.safetensors b/Qwen-Image-13B-8steps/transformer/diffusion_pytorch_model-00003-of-00003.safetensors similarity index 100% rename from Qwen-Image-13B/transformer/diffusion_pytorch_model-00003-of-00003.safetensors rename to Qwen-Image-13B-8steps/transformer/diffusion_pytorch_model-00003-of-00003.safetensors diff --git a/Qwen-Image-13B/transformer/diffusion_pytorch_model.safetensors.index.json b/Qwen-Image-13B-8steps/transformer/diffusion_pytorch_model.safetensors.index.json similarity index 100% rename from Qwen-Image-13B/transformer/diffusion_pytorch_model.safetensors.index.json rename to Qwen-Image-13B-8steps/transformer/diffusion_pytorch_model.safetensors.index.json diff --git a/Qwen-Image-13B/vae/config.json b/Qwen-Image-13B-8steps/vae/config.json similarity index 100% rename from Qwen-Image-13B/vae/config.json rename to Qwen-Image-13B-8steps/vae/config.json diff --git a/Qwen-Image-13B/vae/diffusion_pytorch_model.safetensors b/Qwen-Image-13B-8steps/vae/diffusion_pytorch_model.safetensors similarity index 100% rename from Qwen-Image-13B/vae/diffusion_pytorch_model.safetensors rename to Qwen-Image-13B-8steps/vae/diffusion_pytorch_model.safetensors