Antoine Chaffin commited on
Commit
d4061f1
1 Parent(s): aab4f47

Tuning WM parameters

Browse files
Files changed (1) hide show
  1. watermark.py +2 -2
watermark.py CHANGED
@@ -59,8 +59,8 @@ class WatermarkingAaronsonLogitsProcessor( WatermarkingLogitsProcessor):
59
 
60
  class WatermarkingKirchenbauerLogitsProcessor(WatermarkingLogitsProcessor):
61
  def __init__(self, *args,
62
- gamma = 0.5,
63
- delta = 4.0,
64
  **kwargs):
65
  super().__init__(*args, **kwargs)
66
  self.gamma = gamma
 
59
 
60
  class WatermarkingKirchenbauerLogitsProcessor(WatermarkingLogitsProcessor):
61
  def __init__(self, *args,
62
+ gamma = 0.25,
63
+ delta = 5.0,
64
  **kwargs):
65
  super().__init__(*args, **kwargs)
66
  self.gamma = gamma