Studio Images: keep curated safetensors models in Recommended after download

The curated bnb-4bit / fp8 diffusion rows were filtered out of the Images picker's
Recommended list once cached (curatedSafetensorsRows dropped anything in downloadedSet),
so they vanished from the picker after the first load and could only be found by typing an
exact search. The row already renders a downloaded badge, matching how GGUF Recommended
rows stay visible when cached. Drop the exclusion so the curated safetensors always list.
This commit is contained in:
Daniel Han 2026-07-01 06:35:14 +00:00
commit c42efd87ec

View file

@ -1690,10 +1690,12 @@ export function HubModelPicker({
// curates them; already-downloaded ones show under Downloaded instead.
const curatedSafetensorsRows = useMemo(() => {
if (!task) return [];
return models.filter(
(m) => m.isGguf === false && !downloadedSet.has(m.id.toLowerCase()),
);
}, [models, task, downloadedSet]);
// Always list the curated safetensors (bnb-4bit / fp8) diffusion models. They
// render with a "downloaded" badge when cached (like GGUF Recommended rows), so
// they must not be hidden once on disk -- otherwise they vanish from the picker
// entirely after the first load.
return models.filter((m) => m.isGguf === false);
}, [models, task]);
// Per-row meta + VRAM badge from the recommended listing's own metadata.
const recommendedMeta = useMemo(() => {