Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Yingda Chen
2024-11-01 18:12:58 +08:00
parent e2b55397f9
commit 147b764f36
1 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
modelscope/utils/hub.py
View File

@@ -22,15 +22,9 @@ def create_model_if_not_exist(
model_id: str,
chinese_name: str,
visibility: Optional[int] = ModelVisibility.PUBLIC,
license: Optional[str] = Licenses.APACHE_V2,
revision: Optional[str] = DEFAULT_MODEL_REVISION):
exists = True
try:
api.get_model(model_id=model_id, revision=revision)
except HTTPError:
exists = False
if exists:
print(f'model {model_id} already exists, skip creation.')
license: Optional[str] = Licenses.APACHE_V2):
if api.repo_exists(model_id):
logger.info(f'model {model_id} already exists, skip creation.')
return False
else:
api.create_model(
@@ -39,7 +33,7 @@ def create_model_if_not_exist(
license=license,
chinese_name=chinese_name,
)
print(f'model {model_id} successfully created.')
logger.info(f'model {model_id} successfully created.')
return True