removed progressbar

main/api.py

@@ -58,11 +58,13 @@ class LLMApi:
             model_path = self.models_path / model_name.split('/')[-1]
 
             # Download and save model
+            self.logger.info(f"Enabling stdout logging for download")
             self.logger.enable_stream_to_logger()
-            model = AutoModelForCausalLM.from_pretrained(model_name, progress_bar=True)
+            model = AutoModelForCausalLM.from_pretrained(model_name)
 
             # Download and save tokenizer
-            tokenizer = AutoTokenizer.from_pretrained(model_name, progress_bar=True)
+            tokenizer = AutoTokenizer.from_pretrained(model_name)
+            self.logger.info(f"Disnabling stdout logging")
             self.logger.disable_stream_to_logger()
 
             self.logger.info(f"Saving model to {model_path}")