Commit
•
cad3946
1
Parent(s):
2d1b51f
downloads AVAILABLE MODELS instead of some random model when running
Browse files- download_repo.py +4 -1
download_repo.py
CHANGED
@@ -2,6 +2,8 @@ import urllib3
|
|
2 |
|
3 |
from huggingface_hub import snapshot_download
|
4 |
|
|
|
|
|
5 |
|
6 |
def change_default_timeout(new_timeout: int) -> None:
|
7 |
"""
|
@@ -39,4 +41,5 @@ def download_pytorch_model(name: str) -> None:
|
|
39 |
|
40 |
|
41 |
if __name__ == "__main__":
|
42 |
-
|
|
|
|
2 |
|
3 |
from huggingface_hub import snapshot_download
|
4 |
|
5 |
+
from available_models import AVAILABLE_MODELS
|
6 |
+
|
7 |
|
8 |
def change_default_timeout(new_timeout: int) -> None:
|
9 |
"""
|
|
|
41 |
|
42 |
|
43 |
if __name__ == "__main__":
|
44 |
+
for model_name in AVAILABLE_MODELS:
|
45 |
+
download_pytorch_model(model_name)
|