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

fixing the task filtering

Browse source
This commit is contained in:
Manan17 2026-03-17 04:52:21 +00:00
commit e8706ba362
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
studio/frontend/src/hooks/use-hf-model-search.ts
View file

@ -176,7 +176,9 @@ async function* priorityThenListingIterator(
);
for (const result of settled) {
if (result.status === "fulfilled") {
const m = result.value as { name?: string };
const m = result.value as { name?: string; pipeline_tag?: string };
// Skip models that don't match the selected task filter
if (task && m.pipeline_tag && m.pipeline_tag !== task) continue;
if (m.name) seen.add(m.name);
yield result.value;
}