Studio Images: list on-device unsloth diffusion models in the picker
The Images picker's On Device tab hid every non-GGUF cached repo whenever a task filter was active, so downloaded unsloth diffusion pipelines (bnb-4bit and FP8 safetensors) never showed up there. List cached repos that pass the task gate, limited under a filter to unsloth-hosted ones so base repos (which fail the diffusion load trust gate) don't appear only to dead-end on click. Chat behavior is unchanged: the task gate still drops image repos there.
This commit is contained in:
parent
38a69ab154
commit
604949390b
1 changed files with 13 additions and 10 deletions
|
|
@ -1790,18 +1790,21 @@ export function HubModelPicker({
|
|||
),
|
||||
[cachedGguf, downloadedSort, loadTimes, task],
|
||||
);
|
||||
// Non-GGUF (safetensors) cached repos aren't single-file diffusion GGUFs, so
|
||||
// hide them entirely when a task filter is active (the Images picker). In chat,
|
||||
// drop cached diffusers pipeline repos (a Z-Image / FLUX base) the same way.
|
||||
// Cached non-GGUF repos. In chat, passesTaskGate drops diffusers image repos. In the
|
||||
// Images picker (task set) it keeps them, but limit to repos this backend can actually
|
||||
// load as diffusion: unsloth-hosted ones. Base repos (Qwen/Qwen-Image, FLUX bases) are
|
||||
// cached as dependencies and fail the diffusion trust gate, so listing them would dead-end.
|
||||
const sortedCachedModels = useMemo(
|
||||
() =>
|
||||
task
|
||||
? []
|
||||
: sortCachedRepos(
|
||||
cachedModels.filter((c) => passesTaskGate(c.task, c.repo_id, task)),
|
||||
downloadedSort,
|
||||
loadTimes,
|
||||
),
|
||||
sortCachedRepos(
|
||||
cachedModels.filter(
|
||||
(c) =>
|
||||
passesTaskGate(c.task, c.repo_id, task) &&
|
||||
(!task || isUnslothRepoId(c.repo_id)),
|
||||
),
|
||||
downloadedSort,
|
||||
loadTimes,
|
||||
),
|
||||
[cachedModels, downloadedSort, loadTimes, task],
|
||||
);
|
||||
// Each local section's search is scoped to its own models (matched by name).
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue