diff --git a/src/pages/llmodels/components/table-list.tsx b/src/pages/llmodels/components/table-list.tsx index 40e5d789..4a626cb4 100644 --- a/src/pages/llmodels/components/table-list.tsx +++ b/src/pages/llmodels/components/table-list.tsx @@ -222,7 +222,12 @@ const Models: React.FC = ({ const handleModalOk = useCallback( async (data: FormData) => { try { - await updateModel({ data, id: currentData?.id as number }); + await updateModel({ + data: { + ...data + }, + id: currentData?.id as number + }); setOpenAddModal(false); message.success(intl.formatMessage({ id: 'common.message.success' })); } catch (error) {} diff --git a/src/pages/llmodels/components/update-modal.tsx b/src/pages/llmodels/components/update-modal.tsx index 181a601e..82ed324e 100644 --- a/src/pages/llmodels/components/update-modal.tsx +++ b/src/pages/llmodels/components/update-modal.tsx @@ -247,17 +247,21 @@ const UpdateModal: React.FC = (props) => { }; const handleOk = (formdata: FormData) => { - const gpu = _.find(gpuOptions, (item: any) => { - return item.value === formdata.gpu_selector; - }); - if (gpu) { + if (formdata.scheduleType === 'manual') { + const gpu = _.find(gpuOptions, (item: any) => { + return item.value === formdata.gpu_selector; + }); + onOk({ ..._.omit(formdata, ['scheduleType']), - gpu_selector: { - gpu_name: gpu.name, - gpu_index: gpu.index, - worker_name: gpu.worker_name - } + worker_selector: null, + gpu_selector: gpu + ? { + gpu_name: gpu.name, + gpu_index: gpu.index, + worker_name: gpu.worker_name + } + : null }); } else { onOk({