diff --git a/src/pages/llmodels/components/advance-config.tsx b/src/pages/llmodels/components/advance-config.tsx index fd3c33ad..86c22f83 100644 --- a/src/pages/llmodels/components/advance-config.tsx +++ b/src/pages/llmodels/components/advance-config.tsx @@ -365,6 +365,7 @@ const AdvanceConfig: React.FC = (props) => { ]; }, [ form, + source, intl, gpuOptions, paramsConfig, diff --git a/src/pages/llmodels/components/data-form.tsx b/src/pages/llmodels/components/data-form.tsx index c5962cc4..377c90c4 100644 --- a/src/pages/llmodels/components/data-form.tsx +++ b/src/pages/llmodels/components/data-form.tsx @@ -102,6 +102,9 @@ const DataForm: React.FC = forwardRef((props, ref) => { }, getFieldValue: (name: string) => { return form.getFieldValue(name); + }, + resetFields() { + form.resetFields(); } }; }, @@ -129,11 +132,12 @@ const DataForm: React.FC = forwardRef((props, ref) => { const handleLocalPathBlur = (e: any) => { const value = e.target.value; const isEndwithGGUF = _.endsWith(value, '.gguf'); - if (isEndwithGGUF) { - props.onBackendChange?.(backendOptionsMap.llamaBox); - } else { - props.onBackendChange?.(backendOptionsMap.vllm); + let backend = backendOptionsMap.llamaBox; + if (!isEndwithGGUF) { + backend = backendOptionsMap.vllm; } + props.onBackendChange?.(backend); + form.setFieldValue('backend', backend); }; const renderHuggingfaceFields = () => { diff --git a/src/pages/llmodels/components/deploy-modal.tsx b/src/pages/llmodels/components/deploy-modal.tsx index da5d120e..9caa23ea 100644 --- a/src/pages/llmodels/components/deploy-modal.tsx +++ b/src/pages/llmodels/components/deploy-modal.tsx @@ -4,7 +4,7 @@ import { CloseOutlined } from '@ant-design/icons'; import { useIntl } from '@umijs/max'; import { Button, Drawer } from 'antd'; import { debounce } from 'lodash'; -import { memo, useCallback, useEffect, useRef, useState } from 'react'; +import { memo, useCallback, useEffect, useId, useRef, useState } from 'react'; import { backendOptionsMap, modelSourceMap } from '../config'; import { FormData, ListItem } from '../config/types'; import ColumnWrapper from './column-wrapper'; @@ -40,6 +40,7 @@ const AddModal: React.FC = (props) => { modelSourceMap.huggingface_value, modelSourceMap.modelscope_value ]; + const uid = useId(); const form = useRef({}); const intl = useIntl(); const [selectedModel, setSelectedModel] = useState({}); @@ -81,6 +82,11 @@ const AddModal: React.FC = (props) => { } }; + const handleCancel = useCallback(() => { + onCancel?.(); + setIsGGUF(false); + }, [onCancel]); + useEffect(() => { handleSelectModelFile({ fakeName: '' }); }, [selectedModel]); @@ -110,13 +116,13 @@ const AddModal: React.FC = (props) => { > {title} - } open={open} - onClose={onCancel} + onClose={handleCancel} destroyOnClose={true} closeIcon={false} maskClosable={false} @@ -173,7 +179,7 @@ const AddModal: React.FC = (props) => {