Fix dependency (#1085)

* fix yaml dependency for hub

---------

Co-authored-by: Yingda Chen <yingda.chen@alibaba-inc.com>
(cherry picked from commit 91d85f23c2)
This commit is contained in:
Yingda Chen
2024-11-14 16:41:24 +08:00
committed by yuze.zyz
parent 9f3529e1f8
commit de04baf070

View File

@@ -22,8 +22,6 @@ import requests
from requests import Session
from requests.adapters import HTTPAdapter, Retry
from modelscope import utils
from modelscope.fileio import io
from modelscope.hub.constants import (API_HTTP_CLIENT_MAX_RETRIES,
API_HTTP_CLIENT_TIMEOUT,
API_RESPONSE_FIELD_DATA,
@@ -300,7 +298,8 @@ class HubApi:
ConfigFields.framework: Frameworks.torch,
ConfigFields.task: Tasks.other,
}
io.dump(cfg, cfg_file)
with open(cfg_file, 'w') as file:
json.dump(cfg, file)
def push_model(self,
model_id: str,