Skip to content

Commit b08accd

Browse files
feat(starter-models): add Z-Image Turbo starter models
Add Z-Image Turbo and related models to the starter models list: - Z-Image Turbo (full precision, ~13GB) - Z-Image Turbo quantized (GGUF Q4_K, ~4GB) - Z-Image Qwen3 Text Encoder (full precision, ~8GB) - Z-Image Qwen3 Text Encoder quantized (GGUF Q6_K, ~3.3GB) - Z-Image ControlNet Union (Canny, HED, Depth, Pose, MLSD, Inpainting) The quantized Turbo model includes the quantized Qwen3 encoder as a dependency for automatic installation.
1 parent ab6b672 commit b08accd

File tree

1 file changed

+50
-0
lines changed

1 file changed

+50
-0
lines changed

invokeai/backend/model_manager/starter_models.py

Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -690,6 +690,51 @@ class StarterModelBundle(BaseModel):
690690
)
691691
# endregion
692692

693+
# region Z-Image
694+
z_image_qwen3_encoder = StarterModel(
695+
name="Z-Image Qwen3 Text Encoder",
696+
base=BaseModelType.Any,
697+
source="Tongyi-MAI/Z-Image-Turbo::text_encoder",
698+
description="Qwen3 4B text encoder for Z-Image (full precision). ~8GB",
699+
type=ModelType.Qwen3Encoder,
700+
)
701+
702+
z_image_qwen3_encoder_quantized = StarterModel(
703+
name="Z-Image Qwen3 Text Encoder (quantized)",
704+
base=BaseModelType.Any,
705+
source="https://huggingface.co/worstplayer/Z-Image_Qwen_3_4b_text_encoder_GGUF/resolve/main/Qwen_3_4b-Q6_K.gguf",
706+
description="Qwen3 4B text encoder for Z-Image quantized to GGUF Q6_K format. ~3.3GB",
707+
type=ModelType.Qwen3Encoder,
708+
format=ModelFormat.GGUFQuantized,
709+
)
710+
711+
z_image_turbo = StarterModel(
712+
name="Z-Image Turbo",
713+
base=BaseModelType.ZImage,
714+
source="Tongyi-MAI/Z-Image-Turbo",
715+
description="Z-Image Turbo - fast 6B parameter text-to-image model with 8 inference steps. Supports bilingual prompts (English & Chinese). ~13GB",
716+
type=ModelType.Main,
717+
)
718+
719+
z_image_turbo_quantized = StarterModel(
720+
name="Z-Image Turbo (quantized)",
721+
base=BaseModelType.ZImage,
722+
source="https://huggingface.co/leejet/Z-Image-Turbo-GGUF/resolve/main/z_image_turbo-Q4_K.gguf",
723+
description="Z-Image Turbo quantized to GGUF Q4_K format. Requires separate Qwen3 text encoder. ~4GB",
724+
type=ModelType.Main,
725+
format=ModelFormat.GGUFQuantized,
726+
dependencies=[z_image_qwen3_encoder_quantized],
727+
)
728+
729+
z_image_controlnet_union = StarterModel(
730+
name="Z-Image ControlNet Union",
731+
base=BaseModelType.ZImage,
732+
source="https://huggingface.co/alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.1/resolve/main/Z-Image-Turbo-Fun-Controlnet-Union-2.1-8steps.safetensors",
733+
description="Unified ControlNet for Z-Image Turbo supporting Canny, HED, Depth, Pose, MLSD, and Inpainting modes.",
734+
type=ModelType.ControlNet,
735+
)
736+
# endregion
737+
693738
# List of starter models, displayed on the frontend.
694739
# The order/sort of this list is not changed by the frontend - set it how you want it here.
695740
STARTER_MODELS: list[StarterModel] = [
@@ -766,6 +811,11 @@ class StarterModelBundle(BaseModel):
766811
cogview4,
767812
flux_krea,
768813
flux_krea_quantized,
814+
z_image_turbo,
815+
z_image_turbo_quantized,
816+
z_image_qwen3_encoder,
817+
z_image_qwen3_encoder_quantized,
818+
z_image_controlnet_union,
769819
]
770820

771821
sd1_bundle: list[StarterModel] = [

0 commit comments

Comments
 (0)