diff --git a/src/pages/resources/components/model-files.tsx b/src/pages/resources/components/model-files.tsx index 39a494c9..48f8a372 100644 --- a/src/pages/resources/components/model-files.tsx +++ b/src/pages/resources/components/model-files.tsx @@ -121,6 +121,28 @@ const ModelFiles = () => { return isGGUF || isOllama; }; + const getModelInfo = (record: ListItem) => { + if (record.source === modelSourceMap.huggingface_value) { + return { + source: modelSourceMap.huggingface_value, + huggingface_repo_id: record.huggingface_repo_id, + huggingface_filename: record.huggingface_filename + }; + } + + if (record.source === modelSourceMap.modelscope_value) { + return { + source: modelSourceMap.modelscope_value, + model_scope_model_id: record.model_scope_model_id, + model_scope_file_path: record.model_scope_file_path + }; + } + return { + source: modelSourceMap.local_path_value, + local_path: record.resolved_paths?.[0] + }; + }; + const generateInitialValues = (record: ListItem, gpuOptions: any[]) => { const isGGUF = checkIsGGUF(record); const audioModelTag = identifyModelTask( @@ -146,8 +168,7 @@ const ModelFiles = () => { cluster_id: workersList?.find( (worker) => worker.value === record.worker_id )?.cluster_id, - source: modelSourceMap.local_path_value, - local_path: record.resolved_paths?.[0], + ...getModelInfo(record), worker_selector: targetWorker ? { 'worker-name': targetWorker