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

feat: sort unsloth models first in HF search dropdowns

Browse source
This commit is contained in:
imagineer99 2026-02-22 01:45:51 +00:00
commit 5bdec309e6
1 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,6 @@
import type { PipelineType } from "@huggingface/hub";
import { listModels } from "@huggingface/hub";
import { useCallback } from "react";
import { useCallback, useMemo } from "react";
import { useHfPaginatedSearch } from "./use-hf-paginated-search";
export interface HfModelResult {
@ -84,5 +84,17 @@ export function useHfModelSearch(
[query, task, accessToken],
);
return useHfPaginatedSearch(createIter, mapModel);
const search = useHfPaginatedSearch(createIter, mapModel);
const results = useMemo(
() =>
[...search.results].sort((a, b) => {
const aFirst = a.id.startsWith("unsloth/") ? 0 : 1;
const bFirst = b.id.startsWith("unsloth/") ? 0 : 1;
return aFirst - bFirst;
}),
[search.results],
);
return { ...search, results };
}