This commit is contained in:
suluyan
2025-12-10 11:09:03 +08:00
parent 9e463b0403
commit b9ac390434

View File

@@ -2912,14 +2912,11 @@ class HubApi:
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')
else:
model_tasks = ''
else:
model_tasks = ''
payload = {
'ChineseName': model_info.get('ChineseName', ''),
'ModelFramework': model_info.get('ModelFramework', 'Pytorch'),