remove logs

nutanix · Feb 27, 2024 · 13c47d9 · 13c47d9
1 parent 334eb15
commit 13c47d9
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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: