From 147b764f36cbd83a87d563dee139610bd2fea894 Mon Sep 17 00:00:00 2001
From: Yingda Chen <yingda.chen@alibaba-inc.com>
Date: Fri, 1 Nov 2024 18:12:58 +0800
Subject: [PATCH] update

---
 modelscope/utils/hub.py | 14 ++++----------
 1 file changed, 4 insertions(+), 10 deletions(-)

diff --git a/modelscope/utils/hub.py b/modelscope/utils/hub.py
index 20fb0e20..39ca644a 100644
--- a/modelscope/utils/hub.py
+++ b/modelscope/utils/hub.py
@@ -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