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

Studio Images: load non-curated unsloth/on-device diffusers repos instead of no-op

Browse source 
handleModelSelect only loaded curated safetensors ids and GGUF variant picks; any other
non-GGUF pick (an on-device diffusers folder, or a future unsloth diffusers image repo
surfaced by search) silently did nothing. Treat such a pick as a full diffusers pipeline
load when the id is unsloth-hosted or on-device (the backend infers the family + base repo
and gates loads to unsloth/* or local paths), and show a clear message otherwise instead
of silently ignoring the click. Curated and GGUF paths are unchanged.
This commit is contained in:
Daniel Han 2026-07-01 06:04:37 +00:00
commit 300f261d35
1 changed files with 20 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

23
studio/frontend/src/features/images/images-page.tsx
View file

@ -1192,12 +1192,29 @@ export function ImagesPage() {
void handleLoad(id, { kind: spec.kind, filename: spec.filename });
return;
}
if (!meta.ggufVariant || !meta.ggufFilename) return; // not a quant pick
setQuant(meta.ggufVariant);
// GGUF quant pick from the variant expander.
if (meta.ggufVariant && meta.ggufFilename) {
setQuant(meta.ggufVariant);
const dq = defaultsFor(id);
setSteps(dq.steps);
setGuidance(dq.guidance);
void handleLoad(id, { kind: "gguf", filename: meta.ggufFilename });
return;
}
// A direct local .gguf file picked without a variant isn't wired for Images.
if (meta.isGguf) return;
// Otherwise treat it as a full diffusers repo (safetensors / bnb-4bit). The backend
// infers the family + base repo from the id and gates loads to unsloth/* repos or
// on-device paths, so only attempt those; other Hub orgs can't be assembled here.
if (meta.source !== "local" && !id.toLowerCase().startsWith("unsloth/")) {
toast.error("Only unsloth or on-device image models can be loaded here");
return;
}
setQuant(null);
const d = defaultsFor(id);
setSteps(d.steps);
setGuidance(d.guidance);
void handleLoad(id, { kind: "gguf", filename: meta.ggufFilename });
void handleLoad(id, { kind: "pipeline" });
},
[handleLoad],
);