yonikremer
commited on
Commit
•
24fe512
1
Parent(s):
de0f77a
renaming
Browse files- on_server_start.py +2 -2
on_server_start.py
CHANGED
@@ -6,7 +6,7 @@ from concurrent.futures import ThreadPoolExecutor
|
|
6 |
from transformers import logging as transformers_logging
|
7 |
from huggingface_hub import logging as huggingface_hub_logging
|
8 |
|
9 |
-
from download_repo import
|
10 |
|
11 |
|
12 |
def disable_progress_bar():
|
@@ -28,7 +28,7 @@ def download_useful_models():
|
|
28 |
"facebook/opt-iml-max-30b",
|
29 |
)
|
30 |
with ThreadPoolExecutor() as executor:
|
31 |
-
executor.map(
|
32 |
|
33 |
|
34 |
async def main():
|
|
|
6 |
from transformers import logging as transformers_logging
|
7 |
from huggingface_hub import logging as huggingface_hub_logging
|
8 |
|
9 |
+
from download_repo import download_pytorch_model
|
10 |
|
11 |
|
12 |
def disable_progress_bar():
|
|
|
28 |
"facebook/opt-iml-max-30b",
|
29 |
)
|
30 |
with ThreadPoolExecutor() as executor:
|
31 |
+
executor.map(download_pytorch_model, useful_models)
|
32 |
|
33 |
|
34 |
async def main():
|