renaming

on_server_start.py

@@ -6,7 +6,7 @@ from concurrent.futures import ThreadPoolExecutor
 from transformers import logging as transformers_logging
 from huggingface_hub import logging as huggingface_hub_logging
 
-from download_repo import download_repository
+from download_repo import download_pytorch_model
 
 
 def disable_progress_bar():
@@ -28,7 +28,7 @@ def download_useful_models():
         "facebook/opt-iml-max-30b",
     )
     with ThreadPoolExecutor() as executor:
-        executor.map(download_repository, useful_models)
+        executor.map(download_pytorch_model, useful_models)
 
 
 async def main():