diff --git a/studio/frontend/src/features/hub/hooks/use-hub-model-search.ts b/studio/frontend/src/features/hub/hooks/use-hub-model-search.ts index 4e7be09f2c..f2d072fc6c 100644 --- a/studio/frontend/src/features/hub/hooks/use-hub-model-search.ts +++ b/studio/frontend/src/features/hub/hooks/use-hub-model-search.ts @@ -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; }