diff --git a/src/pages/llmodels/components/data-form.tsx b/src/pages/llmodels/components/data-form.tsx index f0e4d8aa..625efd55 100644 --- a/src/pages/llmodels/components/data-form.tsx +++ b/src/pages/llmodels/components/data-form.tsx @@ -158,6 +158,9 @@ const DataForm: React.FC = forwardRef((props, ref) => { return { form: form, submit: handleSumit, + resetFields: (fields: any[]) => { + form.resetFields(fields); + }, setFieldsValue: (values: FormData) => { form.setFieldsValue(values); }, diff --git a/src/pages/llmodels/components/deploy-modal.tsx b/src/pages/llmodels/components/deploy-modal.tsx index 2a76f6b5..1a79db8f 100644 --- a/src/pages/llmodels/components/deploy-modal.tsx +++ b/src/pages/llmodels/components/deploy-modal.tsx @@ -23,6 +23,14 @@ import SearchModel from './search-model'; import Separator from './separator'; import TitleWrapper from './title-wrapper'; +const resetFields = [ + 'cpu_offloading', + 'distributed_inference_across_workers', + 'backend', + 'backend_version', + 'backend_parameters' +]; + const ModalFooterStyle = { padding: '16px 24px', display: 'flex', @@ -93,6 +101,7 @@ const AddModal: FC = (props) => { const modelFileRef = useRef(null); const handleSelectModelFile = (item: any) => { + form.current?.form?.resetFields(resetFields); const modelInfo = onSelectModel(selectedModel, props.source); form.current?.setFieldsValue?.({ file_name: item.fakeName, @@ -109,6 +118,7 @@ const AddModal: FC = (props) => { setSelectedModel(item); if (!item.isGGUF) { setIsGGUF(false); + form.current?.form?.resetFields(resetFields); const modelInfo = onSelectModel(item, props.source); handleShowCompatibleAlert(item.evaluateResult); form.current?.setFieldsValue?.({