Spaces:
Runtime error
Runtime error
Antoine Chaffin
commited on
Commit
•
d4061f1
1
Parent(s):
aab4f47
Tuning WM parameters
Browse files- 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.
|
63 |
-
delta =
|
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
|