Remove logging

predibase · Nov 11, 2024 · 530bc44 · 530bc44
1 parent dfb567d
commit 530bc44
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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):