Add FLUX.2-klein-9B to the curated image models

This commit is contained in:
oobabooga 2026-06-24 22:31:57 -03:00
commit 818ddb828f
2 changed files with 3 additions and 0 deletions

View file

@ -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.

View file

@ -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,