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

fixing update model type

Browse source
This commit is contained in:
Manan17 2026-03-06 07:44:57 +00:00
commit cb3f3f4d0c
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

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