downloads AVAILABLE MODELS instead of some random model when running

download_repo.py

@@ -2,6 +2,8 @@ import urllib3
 
 from huggingface_hub import snapshot_download
 
+from available_models import AVAILABLE_MODELS
+
 
 def change_default_timeout(new_timeout: int) -> None:
     """
@@ -39,4 +41,5 @@ def download_pytorch_model(name: str) -> None:
 
 
 if __name__ == "__main__":
-    download_pytorch_model("facebook/opt-125m")
+    for model_name in AVAILABLE_MODELS:
+        download_pytorch_model(model_name)