From 818ddb828fe65ed5254281005a8d56d188f06536 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 24 Jun 2026 22:31:57 -0300 Subject: [PATCH] Add FLUX.2-klein-9B to the curated image models --- studio/backend/tests/test_diffusion_backend.py | 2 ++ studio/frontend/src/features/images/images-page.tsx | 1 + 2 files changed, 3 insertions(+) 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,