Add FLUX.2-klein-9B to the curated image models
This commit is contained in:
parent
6411f9464a
commit
818ddb828f
2 changed files with 3 additions and 0 deletions
|
|
@ -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.
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue