Fixing LLM init v7

main/routes.py

@@ -263,15 +263,15 @@ async def initialize_model(request: InitializeRequest):
         logger.info(f"Using model path: {model_path}")
 
         # Load the model
-        print("Loading model")
+        logger.info("Loading model")
         llm_instance = LLM.load(
             model=model_path,
             distribute=None if request.precision or request.quantize else "auto"
         )
-        print("Done loading model")
+        logger.info("Done loading model")
 
         # If manual distribution is needed
-        print("Distributing model")
+        logger.info("Distributing model")
         if request.precision or request.quantize:
             llm_instance.distribute(
                 accelerator="cuda" if request.mode == "gpu" else "cpu",
@@ -279,7 +279,7 @@ async def initialize_model(request: InitializeRequest):
                 precision=request.precision,
                 quantize=request.quantize
             )
-        print("Done distributing model")
+        logger.info("Done distributing model")
 
         logger.info(
             f"Model initialized successfully with config:\n"