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

feat: infer tts model type from backend is_audio flag

Browse source
This commit is contained in:
imagineer99 2026-03-10 12:57:40 +00:00
commit faf8351825
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
studio/frontend/src/features/training/api/models-api.ts
View file

@ -62,6 +62,7 @@ export interface ModelConfigResponse {
config?: BackendModelConfig | null;
is_vision: boolean;
is_lora: boolean;
is_audio?: boolean;
base_model?: string | null;
model_type?: "text" | "vision" | "tts" | "embeddings" | null;
}

2
studio/frontend/src/features/training/stores/training-config-store.ts
View file

@ -130,7 +130,7 @@ export const useTrainingConfigStore = create<TrainingConfigStore>()(
// Use backend-provided model_type when available, otherwise
// infer from is_vision (temporary until backend ships model_type).
const inferredModelType: ModelType = modelDetails.model_type
?? (modelDetails.is_vision ? "vision" : "text");
?? (modelDetails.is_vision ? "vision" : modelDetails.is_audio ? "tts" : "text");
set({
...patch,