Skip to content

Commit 80be1b7

Browse files
fix: correct inaccurate download size estimates in starter models (invoke-ai#8968)
Verified model sizes against Hugging Face repositories and corrected 11 descriptions that had wrong or outdated download size estimates. Key corrections: - T5-XXL base encoder: ~8GB → ~9.5GB - FLUX.2 VAE: ~335MB → ~168MB (was confused with FLUX.1 VAE) - FLUX.1 Krea dev: ~33GB → ~29GB (uses quantized T5, not full) - FLUX.2 Klein 4B/9B Diffusers: ~10GB/~20GB → ~16GB/~35GB - SD3.5 Medium/Large: ~15GB/~19G → ~16GB/~28GB - CogView4: ~29GB → ~31GB - Z-Image Turbo: ~30.6GB → ~33GB - FLUX.1 Kontext/Krea quantized: ~14GB → ~12GB
1 parent dbbf289 commit 80be1b7

1 file changed

Lines changed: 11 additions & 11 deletions

File tree

invokeai/backend/model_manager/starter_models.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ class StarterModelBundle(BaseModel):
7171
name="t5_base_encoder",
7272
base=BaseModelType.Any,
7373
source="InvokeAI/t5-v1_1-xxl::bfloat16",
74-
description="T5-XXL text encoder (used in FLUX pipelines). ~8GB",
74+
description="T5-XXL text encoder (used in FLUX pipelines). ~9.5GB",
7575
type=ModelType.T5Encoder,
7676
)
7777

@@ -156,39 +156,39 @@ class StarterModelBundle(BaseModel):
156156
name="FLUX.1 Kontext dev (quantized)",
157157
base=BaseModelType.Flux,
158158
source="https://huggingface.co/unsloth/FLUX.1-Kontext-dev-GGUF/resolve/main/flux1-kontext-dev-Q4_K_M.gguf",
159-
description="FLUX.1 Kontext dev quantized (q4_k_m). Total size with dependencies: ~14GB",
159+
description="FLUX.1 Kontext dev quantized (q4_k_m). Total size with dependencies: ~12GB",
160160
type=ModelType.Main,
161161
dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder],
162162
)
163163
flux_krea = StarterModel(
164164
name="FLUX.1 Krea dev",
165165
base=BaseModelType.Flux,
166166
source="https://huggingface.co/InvokeAI/FLUX.1-Krea-dev/resolve/main/flux1-krea-dev.safetensors",
167-
description="FLUX.1 Krea dev. Total size with dependencies: ~33GB",
167+
description="FLUX.1 Krea dev. Total size with dependencies: ~29GB",
168168
type=ModelType.Main,
169169
dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder],
170170
)
171171
flux_krea_quantized = StarterModel(
172172
name="FLUX.1 Krea dev (quantized)",
173173
base=BaseModelType.Flux,
174174
source="https://huggingface.co/InvokeAI/FLUX.1-Krea-dev-GGUF/resolve/main/flux1-krea-dev-Q4_K_M.gguf",
175-
description="FLUX.1 Krea dev quantized (q4_k_m). Total size with dependencies: ~14GB",
175+
description="FLUX.1 Krea dev quantized (q4_k_m). Total size with dependencies: ~12GB",
176176
type=ModelType.Main,
177177
dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder],
178178
)
179179
sd35_medium = StarterModel(
180180
name="SD3.5 Medium",
181181
base=BaseModelType.StableDiffusion3,
182182
source="stabilityai/stable-diffusion-3.5-medium",
183-
description="Medium SD3.5 Model: ~15GB",
183+
description="Medium SD3.5 Model: ~16GB",
184184
type=ModelType.Main,
185185
dependencies=[],
186186
)
187187
sd35_large = StarterModel(
188188
name="SD3.5 Large",
189189
base=BaseModelType.StableDiffusion3,
190190
source="stabilityai/stable-diffusion-3.5-large",
191-
description="Large SD3.5 Model: ~19G",
191+
description="Large SD3.5 Model: ~28GB",
192192
type=ModelType.Main,
193193
dependencies=[],
194194
)
@@ -644,7 +644,7 @@ class StarterModelBundle(BaseModel):
644644
name="CogView4",
645645
base=BaseModelType.CogView4,
646646
source="THUDM/CogView4-6B",
647-
description="The base CogView4 model (~29GB).",
647+
description="The base CogView4 model (~31GB).",
648648
type=ModelType.Main,
649649
)
650650
# endregion
@@ -695,7 +695,7 @@ class StarterModelBundle(BaseModel):
695695
name="FLUX.2 VAE",
696696
base=BaseModelType.Flux2,
697697
source="black-forest-labs/FLUX.2-klein-4B::vae",
698-
description="FLUX.2 VAE (16-channel, same architecture as FLUX.1 VAE). ~335MB",
698+
description="FLUX.2 VAE (16-channel, same architecture as FLUX.1 VAE). ~168MB",
699699
type=ModelType.VAE,
700700
)
701701

@@ -719,7 +719,7 @@ class StarterModelBundle(BaseModel):
719719
name="FLUX.2 Klein 4B (Diffusers)",
720720
base=BaseModelType.Flux2,
721721
source="black-forest-labs/FLUX.2-klein-4B",
722-
description="FLUX.2 Klein 4B in Diffusers format - includes transformer, VAE and Qwen3 encoder. ~10GB",
722+
description="FLUX.2 Klein 4B in Diffusers format - includes transformer, VAE and Qwen3 encoder. ~16GB",
723723
type=ModelType.Main,
724724
)
725725

@@ -745,7 +745,7 @@ class StarterModelBundle(BaseModel):
745745
name="FLUX.2 Klein 9B (Diffusers)",
746746
base=BaseModelType.Flux2,
747747
source="black-forest-labs/FLUX.2-klein-9B",
748-
description="FLUX.2 Klein 9B in Diffusers format - includes transformer, VAE and Qwen3 encoder. ~20GB",
748+
description="FLUX.2 Klein 9B in Diffusers format - includes transformer, VAE and Qwen3 encoder. ~35GB",
749749
type=ModelType.Main,
750750
)
751751

@@ -821,7 +821,7 @@ class StarterModelBundle(BaseModel):
821821
name="Z-Image Turbo",
822822
base=BaseModelType.ZImage,
823823
source="Tongyi-MAI/Z-Image-Turbo",
824-
description="Z-Image Turbo - fast 6B parameter text-to-image model with 8 inference steps. Supports bilingual prompts (English & Chinese). ~30.6GB",
824+
description="Z-Image Turbo - fast 6B parameter text-to-image model with 8 inference steps. Supports bilingual prompts (English & Chinese). ~33GB",
825825
type=ModelType.Main,
826826
)
827827

0 commit comments

Comments
 (0)