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

Merge pull request #323 from unslothai/fix/vision-dataset-search-filter

Browse source 
fixing update model type
This commit is contained in:
Roland Tannous 2026-03-09 00:13:50 +04:00 • committed by GitHub
commit 72ea4c18ef
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
studio/frontend/src/features/studio/sections/dataset-section.tsx
View file

@ -76,6 +76,8 @@ export function DatasetSection() {
setDatasetEvalSplit,
hfToken,
modelType,
isVisionModel,
isCheckingVision,
datasetSliceStart,
setDatasetSliceStart,
datasetSliceEnd,
@ -94,6 +96,8 @@ export function DatasetSection() {
setDatasetEvalSplit: s.setDatasetEvalSplit,
hfToken: s.hfToken,
modelType: s.modelType,
isVisionModel: s.isVisionModel,
isCheckingVision: s.isCheckingVision,
datasetSliceStart: s.datasetSliceStart,
setDatasetSliceStart: s.setDatasetSliceStart,
datasetSliceEnd: s.datasetSliceEnd,
@ -119,6 +123,9 @@ export function DatasetSection() {
}
setInputValue(val);
}
const effectiveModelType = !isCheckingVision && isVisionModel ? "vision" : modelType;
const {
results: hfResults,
isLoading,
@ -126,7 +133,7 @@ export function DatasetSection() {
fetchMore,
error: hfSearchError,
} = useHfDatasetSearch(debouncedQuery, {
modelType,
modelType: effectiveModelType,
accessToken: hfToken || undefined,
});