Comment

predibase · Nov 11, 2024 · b0198a2 · b0198a2
1 parent 530bc44
commit b0198a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/lorax_server/utils/logits_process.py b/server/lorax_server/utils/logits_process.py
@@ -149,6 +149,8 @@ class HeterogeneousFrequencyPenaltyLogitsProcessor(LogitsProcessor):
     Args:
         frequency_penalty (`List[float]`):
             The parameter for frequency penalty. 0.0 means no penalty.
+        presence_penalty (`List[float]`):
+            The parameter for presence penalty. 0.0 means no penalty.
     """
 
     def __init__(