diff --git a/src/pages/llmodels/components/data-form.tsx b/src/pages/llmodels/components/data-form.tsx index 8574b6b2..1a6985c1 100644 --- a/src/pages/llmodels/components/data-form.tsx +++ b/src/pages/llmodels/components/data-form.tsx @@ -126,12 +126,16 @@ const DataForm: React.FC = forwardRef((props, ref) => { let data = null; if (props.source === modelSourceMap.huggingface_value) { data = HuggingFaceModels.find( - (item) => `${item.org}/${item.name}` === props.selectedModel.name + (item) => + `${item.org}/${item.name}`.indexOf(props.selectedModel.name) > -1 || + props.selectedModel.name.indexOf(`${item.org}/${item.name}`) > -1 ); } if (props.source === modelSourceMap.modelscope_value) { data = ModelScopeModels.find( - (item) => `${item.org}/${item.name}` === props.selectedModel.name + (item) => + `${item.org}/${item.name}`.indexOf(props.selectedModel.name) > -1 || + props.selectedModel.name.indexOf(`${item.org}/${item.name}`) > -1 ); } if (data) { diff --git a/src/pages/llmodels/config/audio-catalog.ts b/src/pages/llmodels/config/audio-catalog.ts index 8f0d2667..0a2645d8 100644 --- a/src/pages/llmodels/config/audio-catalog.ts +++ b/src/pages/llmodels/config/audio-catalog.ts @@ -101,5 +101,80 @@ export const ModelScopeModels = [ type: 'tts', org: 'iic', name: 'CosyVoice-300M-SFT' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-large-v2' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-large-v3' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-small.en' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-small' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-medium.en' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-medium' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-tiny' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-distil-whisper-large-v3' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-tiny.en' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-distil-whisper-large-v2' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-base' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-base.en' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-distil-whisper-medium.en' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-distil-whisper-small.en' + }, + { + type: 'stt', + org: 'gpustack', + name: 'faster-whisper-large-v1' } ];