Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin' into release/1.33

Browse Source
This commit is contained in:
suluyan
2025-12-10 11:34:35 +08:00
parent dee3da40e1 95aaca3421
commit ea71904892
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modelscope/hub/api.py
View File

@@ -2911,6 +2911,12 @@ class HubApi:
if repo_type == REPO_TYPE_MODEL:
model_info = self.get_model(model_id=repo_id)
path = f'{self.endpoint}/api/v1/models/{repo_id}'
tasks = model_info.get('Tasks')
model_tasks = ''
if isinstance(tasks, list) and tasks:
first = tasks[0]
if isinstance(first, dict) and first:
model_tasks = first.get('name')
payload = {
'ChineseName': model_info.get('ChineseName', ''),
'ModelFramework': model_info.get('ModelFramework', 'Pytorch'),
@@ -2924,7 +2930,7 @@ class HubApi:
'SubScientificField': model_info.get('SubScientificField', None),
'ScientificField': model_info.get('NEXA', {}).get('ScientificField', ''),
'Source': model_info.get('NEXA', {}).get('Source', ''),
'ModelTask': model_info.get('Tasks', [{}])[0].get('name'),
'ModelTask': model_tasks,
'License': model_info.get('License', ''),
}
elif repo_type == REPO_TYPE_DATASET: