From 13c47d980121b7398724c30275b8070ccea52b0f Mon Sep 17 00:00:00 2001
From: Gavrish Prabhu <gavrish.prabhu@nutanix.com>
Date: Tue, 27 Feb 2024 18:38:15 +0000
Subject: [PATCH] remove logs

---
 llm/handler.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/llm/handler.py b/llm/handler.py
index cabf3cd..b84b97e 100644
--- a/llm/handler.py
+++ b/llm/handler.py
@@ -184,7 +184,6 @@ def preprocess(self, data: List) -> torch.Tensor:
                 self.request["request_type"][idx] = "raw"
                 input_list.append(row_input)
 
-        logger.info("Received text: %s", ", ".join(map(str, input_list)))
         encoded_input = self.tokenizer(input_list, padding=True, return_tensors="pt")[
             "input_ids"
         ].to(self.device)
@@ -227,7 +226,6 @@ def inference(self, data: torch.Tensor, *args, **kwargs) -> List:
 
         inference = []
         inference = self.tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
-        logger.info("Generated text is: %s", ", ".join(map(str, inference)))
         return inference
 
     def postprocess(self, data: List) -> List: