diff --git a/studio/backend/tests/test_diffusion_backend.py b/studio/backend/tests/test_diffusion_backend.py index bd5cb44a16..bc33749d2e 100644 --- a/studio/backend/tests/test_diffusion_backend.py +++ b/studio/backend/tests/test_diffusion_backend.py @@ -40,6 +40,8 @@ def test_detect_family_from_repo_id(): assert klein.name == "flux.2-klein" assert klein.pipeline_class == "Flux2KleinPipeline" assert klein.cfg_kwarg == "guidance_scale" + # Both klein sizes share the one family (base repo resolved per-variant). + assert detect_family("unsloth/FLUX.2-klein-9B-GGUF").name == "flux.2-klein" # Only klein is wired up; the Mistral-based FLUX.2-dev base repo is gated. assert detect_family("unsloth/FLUX.2-dev-GGUF") is None # Qwen-Image guides via true_cfg_scale, not guidance_scale. diff --git a/studio/frontend/src/features/images/images-page.tsx b/studio/frontend/src/features/images/images-page.tsx index 7f6760316a..033762165e 100644 --- a/studio/frontend/src/features/images/images-page.tsx +++ b/studio/frontend/src/features/images/images-page.tsx @@ -74,6 +74,7 @@ const MODELS: ModelOption[] = [ txt2img("unsloth/FLUX.1-schnell-GGUF", "FLUX.1 schnell"), txt2img("unsloth/FLUX.1-dev-GGUF", "FLUX.1 dev"), txt2img("unsloth/FLUX.2-klein-4B-GGUF", "FLUX.2 klein 4B"), + txt2img("unsloth/FLUX.2-klein-9B-GGUF", "FLUX.2 klein 9B"), ]; // Per-model generation defaults (steps + guidance), matched by repo-id substring,