From 530bc44d38e6c114a7affea1e3396d8b1e15e10d Mon Sep 17 00:00:00 2001
From: Travis Addair <travis@predibase.com>
Date: Mon, 11 Nov 2024 12:15:29 -0800
Subject: [PATCH] Remove logging

---
 server/lorax_server/utils/logits_process.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/server/lorax_server/utils/logits_process.py b/server/lorax_server/utils/logits_process.py
index 31ee10e75..1091a87c9 100644
--- a/server/lorax_server/utils/logits_process.py
+++ b/server/lorax_server/utils/logits_process.py
@@ -4,7 +4,6 @@
 from typing import Dict, List, Optional, Union
 
 import torch
-from loguru import logger
 from transformers import (
     LogitsProcessor,
     LogitsWarper,
@@ -175,8 +174,6 @@ def __call__(self, input_ids: torch.Tensor, scores: torch.Tensor) -> torch.Tenso
         scores -= token_freq * self.frequency_penalty_tensor
         scores -= mask * self.presence_penalty_tensor
 
-        logger.info(f"!!! Presence penalty: {self.presence_penalty_tensor} {mask}")
-
         return scores
 
     def filter(self, indices):