ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Studio: do not force diffusers pipelines cross-tagged gguf into the GGUF variant expander

Browse source 
Some diffusers image repos (e.g. unsloth/Qwen-Image-2512-unsloth-bnb-4bit) carry a
stray "gguf" tag on the Hub but ship no .gguf files. The model search classified
them as GGUF from the bare tag, so the picker rendered the GGUF variant expander,
which then dead-ended at "No GGUF variants found." Trust the bare gguf tag only when
the repo is not a diffusers pipeline; the -GGUF name suffix and real gguf metadata
(populated via expand=gguf) remain authoritative, so genuine GGUF repos are unaffected.
This commit is contained in:
Daniel Han 2026-07-01 05:47:58 +00:00
commit 9f39ff74dc
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
studio/frontend/src/features/hub/hooks/use-hub-model-search.ts
View file

@ -217,9 +217,19 @@ function makeMapModel(
) {
return null;
}
// A repo cross-tagged "gguf" but that is actually a diffusers pipeline (e.g.
// an unsloth *-bnb-4bit image model) ships no .gguf files, so the GGUF
// variant expander would dead-end at "No GGUF variants found." Trust the bare
// tag only when the repo is not a diffusers pipeline. The "-GGUF" name suffix
// and real gguf metadata (populated via expand=gguf) stay authoritative.
const isDiffusersPipeline =
m.library_name?.toLowerCase() === "diffusers" ||
Boolean(m.tags?.some((tag) => tag.toLowerCase().startsWith("diffusers:")));
const isGguf =
Boolean(m.tags?.some((tag) => tag.toLowerCase() === "gguf")) ||
isGgufLike(m.name);
isGgufLike(m.name) ||
Boolean(m.gguf) ||
(Boolean(m.tags?.some((tag) => tag.toLowerCase() === "gguf")) &&
!isDiffusersPipeline);
if (excludeGguf && isGguf) {
return null;
}