Added Progressbar

main/api.py

@@ -58,8 +58,11 @@ class LLMApi:
             model_path = self.models_path / model_name.split('/')[-1]
 
             # Download and save model
-            model = AutoModelForCausalLM.from_pretrained(model_name)
-            tokenizer = AutoTokenizer.from_pretrained(model_name)
+            self.logger.info(f"Downloading model: {model_name}")
+            model = AutoModelForCausalLM.from_pretrained(model_name, progress_bar=True)
+
+            self.logger.info(f"Downloading tokenizer for model: {model_name}")
+            tokenizer = AutoTokenizer.from_pretrained(model_name, progress_bar=True)
 
             self.logger.info(f"Saving model to {model_path}")
             model.save_pretrained(model_path)