yonikremer commited on
Commit
24fe512
1 Parent(s): de0f77a
Files changed (1) hide show
  1. 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 download_repository
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(download_repository, useful_models)
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():