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

change the ordering of models

Browse source
This commit is contained in:
Manan17 2026-03-17 04:05:22 +00:00
commit c147775efb
3 changed files with 26 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
studio/frontend/src/config/training.ts
View file

@ -131,3 +131,17 @@ export const MODEL_TYPE_TO_HF_TASK: Record<ModelType, PipelineType> = {
audio: "text-to-speech",
embeddings: "feature-extraction",
};
export const PRIORITY_TRAINING_MODELS: readonly string[] = [
"unsloth/Qwen3.5-2B",
"unsloth/Qwen3.5-9B",
"unsloth/gpt-oss-20b",
"unsloth/NVIDIA-Nemotron-3-Nano-4B",
"unsloth/Qwen3-0.6B",
"unsloth/gemma-3-4b-it",
"unsloth/embeddinggemma-300m",
"unsloth/orpheus-3b-0.1-ft",
"unsloth/Llama-3.1-8B-Instruct",
"unsloth/Llama-3.2-3B-Instruct",
];

9
studio/frontend/src/features/onboarding/components/steps/model-selection-step.tsx
View file

@ -33,7 +33,7 @@ import {
TooltipContent,
TooltipTrigger,
} from "@/components/ui/tooltip";
import { MODEL_TYPE_TO_HF_TASK } from "@/config/training";
import { MODEL_TYPE_TO_HF_TASK, PRIORITY_TRAINING_MODELS } from "@/config/training";
import {
useDebouncedValue,
useGpuInfo,
@ -101,7 +101,12 @@ export function ModelSelectionStep() {
const { error: tokenValidationError, isChecking: isCheckingToken } =
useHfTokenValidation(hfToken);
const resultIds = useMemo(() => hfResults.map((r) => r.id), [hfResults]);
const resultIds = useMemo(() => {
const ids = hfResults.map((r) => r.id);
const prioritySet = new Set(PRIORITY_TRAINING_MODELS);
const rest = ids.filter((id) => !prioritySet.has(id));
return [...PRIORITY_TRAINING_MODELS, ...rest];
}, [hfResults]);
// Match Studio behavior: only show exception signals (OOM/TIGHT) in training flows.
const vramMap = useMemo(() => {

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

@ -28,7 +28,7 @@ import {
TooltipContent,
TooltipTrigger,
} from "@/components/ui/tooltip";
import { MODEL_TYPE_TO_HF_TASK } from "@/config/training";
import { MODEL_TYPE_TO_HF_TASK, PRIORITY_TRAINING_MODELS } from "@/config/training";
import {
useDebouncedValue,
useGpuInfo,
@ -172,7 +172,10 @@ export function ModelSection() {
if (selectedModel && !ids.includes(selectedModel)) {
ids.push(selectedModel);
}
return ids;
const prioritySet = new Set(PRIORITY_TRAINING_MODELS);
const rest = ids.filter((id) => !prioritySet.has(id));
return [...PRIORITY_TRAINING_MODELS, ...rest];
}, [hfResults, selectedModel]);
// Filter out GGUF models — they can't be used for training