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
+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)
});