diff --git a/src/pages/llmodels/components/advance-config.tsx b/src/pages/llmodels/components/advance-config.tsx index 436cd230..a301ada8 100644 --- a/src/pages/llmodels/components/advance-config.tsx +++ b/src/pages/llmodels/components/advance-config.tsx @@ -196,7 +196,19 @@ const AdvanceConfig: React.FC = (props) => { } }; - const handleGpuSelectorChange = (value: any[] | string) => { + const handleBeforeGpuSelectorChange = (gpuIds: any[]) => { + if (backend !== backendOptionsMap.ascendMindie || !gpuIds?.length) { + return; + } + const lastGroupName = gpuIds[gpuIds.length - 1][0]; + + const lastGroupItems = gpuIds.filter((item) => item[0] === lastGroupName); + + form.setFieldValue(['gpu_selector', 'gpu_ids'], lastGroupItems); + }; + + const handleGpuSelectorChange = (value: any[]) => { + handleBeforeGpuSelectorChange(value); onValuesChange?.({}, form.getFieldsValue()); };