made the sever start script asynchronous multithreaded

on_server_start.py

@@ -1,10 +1,13 @@
 """
 A script that is run when the server starts.
 """
-from download_repo import download_repository
+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
+
 
 def disable_progress_bar():
     """
@@ -24,13 +27,11 @@ def download_useful_models():
         "facebook/opt-125m",
         "facebook/opt-iml-max-30b",
     )
-    for model_name in useful_models:
-        download_repository(
-            model_name,
-        )
+    with ThreadPoolExecutor() as executor:
+        executor.map(download_repository, useful_models)
 
 
-def main():
+async def main():
     disable_progress_bar()
     download_useful_models()