diff --git a/src/pages/llmodels/components/data-form.tsx b/src/pages/llmodels/components/data-form.tsx index 625efd55..ac51442c 100644 --- a/src/pages/llmodels/components/data-form.tsx +++ b/src/pages/llmodels/components/data-form.tsx @@ -172,9 +172,6 @@ const DataForm: React.FC = forwardRef((props, ref) => { }, getFieldsValue: () => { return form.getFieldsValue(); - }, - resetFields() { - form.resetFields(); } }; }, @@ -276,6 +273,14 @@ const DataForm: React.FC = forwardRef((props, ref) => { ? false : isGGUF }, + { + label: 'Ascend Mindie', + value: backendOptionsMap.ascendMindie, + disabled: + props.source === modelSourceMap.local_path_value + ? false + : isGGUF + }, { label: 'vox-box', value: backendOptionsMap.voxBox, diff --git a/src/pages/llmodels/components/deploy-builtin-modal.tsx b/src/pages/llmodels/components/deploy-builtin-modal.tsx index 1cb6ce11..fb681b8c 100644 --- a/src/pages/llmodels/components/deploy-builtin-modal.tsx +++ b/src/pages/llmodels/components/deploy-builtin-modal.tsx @@ -51,6 +51,10 @@ const backendOptions = [ label: 'vLLM', value: backendOptionsMap.vllm }, + { + label: 'Ascend Mindie', + value: backendOptionsMap.ascendMindie + }, { label: 'vox-box', value: backendOptionsMap.voxBox @@ -306,12 +310,10 @@ const AddModal: React.FC = (props) => { }; const handleBackendChange = (backend: string) => { - if (backend === backendOptionsMap.vllm) { - setIsGGUF(false); - } - if (backend === backendOptionsMap.llamaBox) { setIsGGUF(true); + } else { + setIsGGUF(false); } const sizeList = handleSetSizeOptions({ backend: backend diff --git a/src/pages/llmodels/components/deploy-modal.tsx b/src/pages/llmodels/components/deploy-modal.tsx index 1a79db8f..0482df56 100644 --- a/src/pages/llmodels/components/deploy-modal.tsx +++ b/src/pages/llmodels/components/deploy-modal.tsx @@ -26,7 +26,6 @@ import TitleWrapper from './title-wrapper'; const resetFields = [ 'cpu_offloading', 'distributed_inference_across_workers', - 'backend', 'backend_version', 'backend_parameters' ]; diff --git a/src/pages/llmodels/components/update-modal.tsx b/src/pages/llmodels/components/update-modal.tsx index 7d07cb90..cfb75f17 100644 --- a/src/pages/llmodels/components/update-modal.tsx +++ b/src/pages/llmodels/components/update-modal.tsx @@ -519,6 +519,14 @@ const UpdateModal: React.FC = (props) => { ? false : isGGUF }, + { + label: 'Ascend Mindie', + value: backendOptionsMap.ascendMindie, + disabled: + formData?.source === modelSourceMap.local_path_value + ? false + : isGGUF + }, { label: 'vox-box', value: backendOptionsMap.voxBox, diff --git a/src/pages/llmodels/config/index.ts b/src/pages/llmodels/config/index.ts index 2149555f..fa8f2833 100644 --- a/src/pages/llmodels/config/index.ts +++ b/src/pages/llmodels/config/index.ts @@ -131,7 +131,8 @@ export const localPathTipsList = [ export const backendOptionsMap = { llamaBox: 'llama-box', vllm: 'vllm', - voxBox: 'vox-box' + voxBox: 'vox-box', + ascendMindie: 'ascend-mindie' }; export const backendParamsHolderTips = {