Skip to content

Commit 111408c

Browse files
feat(mm): add flux krea to starter models
1 parent d7619d4 commit 111408c

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

invokeai/backend/model_manager/starter_models.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,29 @@ class StarterModelBundle(BaseModel):
149149
dependencies=[t5_base_encoder, flux_vae, clip_l_encoder],
150150
)
151151
flux_kontext_quantized = StarterModel(
152-
name="FLUX.1 Kontext dev (Quantized)",
152+
name="FLUX.1 Kontext dev (quantized)",
153153
base=BaseModelType.Flux,
154154
source="https://huggingface.co/unsloth/FLUX.1-Kontext-dev-GGUF/resolve/main/flux1-kontext-dev-Q4_K_M.gguf",
155155
description="FLUX.1 Kontext dev quantized (q4_k_m). Total size with dependencies: ~14GB",
156156
type=ModelType.Main,
157157
dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder],
158158
)
159+
flux_krea = StarterModel(
160+
name="FLUX.1 Krea dev",
161+
base=BaseModelType.Flux,
162+
source="https://huggingface.co/InvokeAI/FLUX.1-Krea-dev/resolve/main/flux1-krea-dev.safetensors",
163+
description="FLUX.1 Krea dev. Total size with dependencies: ~33GB",
164+
type=ModelType.Main,
165+
dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder],
166+
)
167+
flux_krea_quantized = StarterModel(
168+
name="FLUX.1 Krea dev (quantized)",
169+
base=BaseModelType.Flux,
170+
source="https://huggingface.co/InvokeAI/FLUX.1-Krea-dev-GGUF/resolve/main/flux1-krea-dev-Q4_K_M.gguf",
171+
description="FLUX.1 Krea dev quantized (q4_k_m). Total size with dependencies: ~14GB",
172+
type=ModelType.Main,
173+
dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder],
174+
)
159175
sd35_medium = StarterModel(
160176
name="SD3.5 Medium",
161177
base=BaseModelType.StableDiffusion3,
@@ -744,6 +760,8 @@ class StarterModelBundle(BaseModel):
744760
llava_onevision,
745761
flux_fill,
746762
cogview4,
763+
flux_krea,
764+
flux_krea_quantized,
747765
]
748766

749767
sd1_bundle: list[StarterModel] = [
@@ -795,6 +813,7 @@ class StarterModelBundle(BaseModel):
795813
flux_redux,
796814
flux_fill,
797815
flux_kontext_quantized,
816+
flux_krea_quantized,
798817
]
799818

800819
STARTER_BUNDLES: dict[str, StarterModelBundle] = {

0 commit comments

Comments
 (0)