disabled progress bars

on_server_start.py

@@ -4,6 +4,14 @@ A script that is run when the server starts.
 from download_repo import download_repository
 
 
+def disable_progress_bar():
+    """
+    Disables the progress bar when downloading models.
+    """
+    import transformers
+    transformers.logging.disable_progress_bar()
+
+
 def download_useful_models():
     """
     Downloads the models that are useful for this project.
@@ -14,10 +22,14 @@ def download_useful_models():
         "facebook/opt-125m",
     )
     for model_name in useful_models:
-        download_repository(model_name)
+        download_repository(
+            model_name,
+
+        )
 
 
 def main():
+    disable_progress_bar()
     download_useful_models()
 
 

requirements.txt

@@ -10,5 +10,4 @@ accelerate
 bitsandbytes
 pytest~=7.1.2
 sentencepiece
-google-api-python-client
-tqdm==4.64.1
+google-api-python-client