fix: vox-box gpu-selector

This commit is contained in:
jialin
2025-02-21 11:45:50 +08:00
parent f1ab927a7e
commit 11d824a88d
8 changed files with 150872 additions and 18 deletions
@@ -124,7 +124,7 @@ const AdvanceConfig: React.FC<AdvanceConfigProps> = (props) => {
}, []);
const handleGPUSelectorChange = (gpuIds: any[]) => {
console.log('GPU Selector Change:', gpuIds);
// only handle for vllm, pick the last selected group
if (
backend === backendOptionsMap.llamaBox ||
backend === backendOptionsMap.voxBox ||
@@ -135,9 +135,6 @@ const AdvanceConfig: React.FC<AdvanceConfigProps> = (props) => {
const lastGroupName = gpuIds[gpuIds.length - 1][0];
const lastGroupItems = gpuIds.filter((item) => item[0] === lastGroupName);
console.log('Last Group Items:', lastGroupItems);
form.setFieldValue(['gpu_selector', 'gpu_ids'], lastGroupItems);
};
+12 -6
View File
@@ -515,12 +515,18 @@ const DataForm: React.FC<DataFormProps> = forwardRef((props, ref) => {
return {};
}
const result = _.map(gpu_ids, (item: string[][] | string[]) => {
if (Array.isArray(item)) {
return item[1];
}
return item;
});
const result = _.reduce(
gpu_ids,
(acc: string[], item: string | string[], index: number) => {
if (Array.isArray(item)) {
acc.push(item[1]);
} else if (index === 1) {
acc.push(item);
}
return acc;
},
[]
);
if (result.length) {
return {
+12 -6
View File
@@ -379,12 +379,18 @@ const UpdateModal: React.FC<AddModalProps> = (props) => {
return {};
}
const result = _.map(gpu_ids, (item: string[][] | string[]) => {
if (Array.isArray(item)) {
return item[1];
}
return item;
});
const result = _.reduce(
gpu_ids,
(acc: string[], item: string | string[], index: number) => {
if (Array.isArray(item)) {
acc.push(item[1]);
} else if (index === 1) {
acc.push(item);
}
return acc;
},
[]
);
if (result.length) {
return {