From c42efd87ec76e39b324ceeff27949046e20fd6d8 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 1 Jul 2026 06:35:14 +0000 Subject: [PATCH] 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. --- .../components/assistant-ui/model-selector/pickers.tsx | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/studio/frontend/src/components/assistant-ui/model-selector/pickers.tsx b/studio/frontend/src/components/assistant-ui/model-selector/pickers.tsx index aae3d75bff..5fe8c4fd6d 100644 --- a/studio/frontend/src/components/assistant-ui/model-selector/pickers.tsx +++ b/studio/frontend/src/components/assistant-ui/model-selector/pickers.tsx @@ -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(() => {