small util for downloading hf models

fhswf · Sep 24, 2024 · 427c76c · 427c76c
1 parent 10b783c
commit 427c76c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/images/downloadtool/download_model.py b/images/downloadtool/download_model.py
@@ -11,5 +11,9 @@
 if model_id is None:
     print("WARNING: Not all envs where specified, can not download anything")
 else:
-    snapshot_download(repo_id=model_id, local_dir=os.path.join(model_folder, model_id), local_dir_use_symlinks=False, revision="main")
+    path = os.path.join(model_folder, model_id)
+    if len(os.listdir(path)) != 0:
+        print(f"WARNING: Model dir for {model_id} is not empty, skipping download")
+    else:
+        snapshot_download(repo_id=model_id, local_dir=os.path.join(model_folder, model_id), local_dir_use_symlinks=False, revision="main")