fix: omit distribution and cpu_offload fields from default spec

This commit is contained in:
jialin
2025-04-08 20:58:24 +08:00
parent 5269cbf862
commit ade1db5215
10 changed files with 85 additions and 16 deletions
@@ -5,6 +5,7 @@ import {
LoadingOutlined,
WarningFilled
} from '@ant-design/icons';
import { Button } from 'antd';
import { isArray } from 'lodash';
import React, { useMemo } from 'react';
import styled from 'styled-components';
@@ -24,13 +25,19 @@ interface CompatibilityAlertProps {
const DivWrapper = styled.div`
position: relative;
padding-inline: 12px;
padding-inline: 8px;
&:hover {
.close-wrapper {
display: block;
}
}
`;
const CloseWrapper = styled.div`
display: none;
position: absolute;
top: 12px;
right: 18px;
top: 6px;
right: 12px;
line-height: 1;
cursor: pointer;
background-color: var(--ant-color-warning-bg);
@@ -98,8 +105,13 @@ const CompatibilityAlert: React.FC<CompatibilityAlertProps> = (props) => {
icon={renderIcon}
></AlertBlockInfo>
{showClose && !['transition', 'success'].includes(type) && (
<CloseWrapper onClick={onClose}>
<CloseOutlined />
<CloseWrapper className="close-wrapper">
<Button
onClick={onClose}
icon={<CloseOutlined />}
size="small"
type="text"
></Button>
</CloseWrapper>
)}
</DivWrapper>
+10 -2
View File
@@ -99,6 +99,14 @@ const AddModal: FC<AddModalProps> = (props) => {
const [isGGUF, setIsGGUF] = useState<boolean>(props.isGGUF || false);
const modelFileRef = useRef<any>(null);
const getDefaultSpec = (item: any) => {
const defaultSpec = item.evaluateResult?.default_spec || {};
return _.omit(defaultSpec, [
'cpu_offloading',
'distributed_inference_across_workers'
]);
};
const getCategory = (item: any) => {
const categories = item.evaluateResult?.default_spec?.categories || [];
if (Array.isArray(categories)) {
@@ -111,7 +119,7 @@ const AddModal: FC<AddModalProps> = (props) => {
const modelInfo = onSelectModel(selectedModel, props.source);
form.current?.setFieldsValue?.({
file_name: item.fakeName,
...item.evaluateResult?.default_spec,
...getDefaultSpec(item),
...modelInfo,
categories: getCategory(item)
});
@@ -129,7 +137,7 @@ const AddModal: FC<AddModalProps> = (props) => {
const modelInfo = onSelectModel(item, props.source);
handleShowCompatibleAlert(item.evaluateResult);
form.current?.setFieldsValue?.({
...item.evaluateResult?.default_spec,
...getDefaultSpec(item),
...modelInfo,
categories: getCategory(item)
});
@@ -70,6 +70,11 @@ export const HuggingFaceModels = [
type: 'tts',
org: 'FunAudioLLM',
name: 'CosyVoice-300M'
},
{
type: 'tts',
org: 'FunAudioLLM',
name: 'CosyVoice2-0.5B'
}
];
@@ -104,6 +109,11 @@ export const ModelScopeModels = [
org: 'iic',
name: 'CosyVoice-300M-25Hz'
},
{
type: 'tts',
org: 'iic',
name: 'CosyVoice2-0.5B'
},
{
type: 'tts',
org: 'iic',
+38 -1
View File
@@ -62,7 +62,16 @@ export const addWorkerGuide: Record<string, any> = {
}) {
return `docker run -d --name gpustack \
--restart=unless-stopped \
-e ASCEND_VISIBLE_DEVICES=0,1 \
--device /dev/davinci0 \
--device /dev/davinci1 \
--device /dev/davinci_manager \
--device /dev/devmm_svm \
--device /dev/hisi_hdc \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
-v /etc/ascend_install.info:/etc/ascend_install.info \
--network=host \
--ipc=host \
-v gpustack-data:/var/lib/gpustack \
@@ -70,6 +79,34 @@ export const addWorkerGuide: Record<string, any> = {
--server-url ${params.server} --token ${params.token} --worker-ip ${params.workerip}`;
}
},
npu310p: {
getToken:
'Get-Content -Path (Join-Path -Path $env:APPDATA -ChildPath "gpustack\\token") -Raw',
registerWorker(params: {
server: string;
tag: string;
token: string;
workerip: string;
}) {
return `docker run -d --name gpustack \
--restart=unless-stopped \
--device /dev/davinci0 \
--device /dev/davinci1 \
--device /dev/davinci_manager \
--device /dev/devmm_svm \
--device /dev/hisi_hdc \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
-v /etc/ascend_install.info:/etc/ascend_install.info \
--network=host \
--ipc=host \
-v gpustack-data:/var/lib/gpustack \
gpustack/gpustack:${params.tag}-310p \
--server-url ${params.server} --token ${params.token} --worker-ip ${params.workerip}`;
}
},
musa: {
getToken:
'Get-Content -Path (Join-Path -Path $env:APPDATA -ChildPath "gpustack\\token") -Raw',