remove logs

nutanix · Feb 27, 2024 · 44ab83c · 44ab83c
1 parent c114635
commit 44ab83c
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/llm/handler.py b/llm/handler.py
@@ -177,7 +177,6 @@ def preprocess(self, data: str) -> 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)
@@ -220,7 +219,6 @@ def inference(self, data: torch.Tensor, *args, **kwargs) -> List[str]:
 
         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[str]) -> List[str]: