add push to hub tracker (modelscope#1249)

(cherry picked from commit 16cc33c)
kronoszhang · Jun 28, 2024 · 2383259 · 2383259
1 parent d31d889
commit 2383259
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/swift/hub/api.py b/swift/hub/api.py
@@ -9,6 +9,7 @@
 import re
 import shutil
 import tempfile
+import time
 import uuid
 from http import HTTPStatus
 from http.cookiejar import CookieJar
@@ -142,7 +143,7 @@ def create_model(self,
             'Visibility': visibility,  # server check
             'License': license,
             'OriginalModelId': original_model_id,
-            'TrainId': os.environ.get('MODELSCOPE_TRAIN_ID', ''),
+            'TrainId': os.environ.get('MODELSCOPE_TRAIN_ID') or f'swift-{time.time()}',
         }
         r = self.session.post(
             path, json=body, cookies=cookies, headers=self.headers)