fix: omit distribution and cpu_offload fields from default spec
This commit is contained in:
@@ -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>
|
||||
|
||||
@@ -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',
|
||||
|
||||
@@ -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',
|
||||
|
||||
Reference in New Issue
Block a user