waterfall 0.2.9__py3-none-any.whl → 0.2.10__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. waterfall/WatermarkerBase.py +1 -1
  2. {waterfall-0.2.9.dist-info → waterfall-0.2.10.dist-info}/METADATA +2 -2
  3. waterfall-0.2.10.dist-info/RECORD +12 -0
  4. waterfall-0.2.9.dist-info/RECORD +0 -12
  5. {waterfall-0.2.9.dist-info → waterfall-0.2.10.dist-info}/WHEEL +0 -0
  6. {waterfall-0.2.9.dist-info → waterfall-0.2.10.dist-info}/entry_points.txt +0 -0
  7. {waterfall-0.2.9.dist-info → waterfall-0.2.10.dist-info}/licenses/LICENSE +0 -0
waterfall/WatermarkerBase.py CHANGED
@@ -280,7 +280,7 @@ class Watermarker:
280
280
  if num_beams <= 1:
281
281
  kwargs["diversity_penalty"] = None
282
282
 
283
- if num_beams > 1 and temperature != 1.0:
283
+ if num_beams > 1 and temperature is not None and temperature != 1.0:
284
284
  logits_processor.append(TemperatureLogitsWarper(float(temperature)))
285
285
  if top_k is not None and top_k != 0:
286
286
  logits_processor.append(TopKLogitsWarper(top_k))
{waterfall-0.2.9.dist-info → waterfall-0.2.10.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: waterfall
3
- Version: 0.2.9
3
+ Version: 0.2.10
4
4
  Summary: Scalable Framework for Robust Text Watermarking and Provenance for LLMs
5
5
  Project-URL: Homepage, https://github.com/aoi3142/Waterfall
6
6
  Project-URL: Issues, https://github.com/aoi3142/Waterfall/issues
@@ -15,7 +15,7 @@ Requires-Dist: numpy>=1.25.0
15
15
  Requires-Dist: scipy>=1.13.0
16
16
  Requires-Dist: sentence-transformers>=3.0.0
17
17
  Requires-Dist: torch>=2.3.0
18
- Requires-Dist: transformers<4.55.0,>=4.43.1
18
+ Requires-Dist: transformers<4.57.0,>=4.43.1
19
19
  Description-Content-Type: text/markdown
20
20
 
21
21
  # Waterfall: Scalable Framework for Robust Text Watermarking and Provenance for LLMs [EMNLP 2024 Main Long]
waterfall-0.2.10.dist-info/RECORD ADDED
@@ -0,0 +1,12 @@
1
+ waterfall/WatermarkerBase.py,sha256=1IvGo1rz1Ec-NW8rQ9bSC8KNdHawu4gl4CNsNncce7Q,22046
2
+ waterfall/WatermarkingFn.py,sha256=-b-kGRdL0a7eKRqJmcHPAR_rCjxQYnsg1Ne6bTwBc1I,1931
3
+ waterfall/WatermarkingFnFourier.py,sha256=QYayAQYwi1dQkDIyqmvhU568VhrVYTVy47HkI8F8SZs,1358
4
+ waterfall/WatermarkingFnSquare.py,sha256=2PAO05DdKT02npo7GDf_82D520nP7kGAWK6H4E4JMt4,1638
5
+ waterfall/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
+ waterfall/permute.py,sha256=uYKdmn4pGvjB6hInInLGxFIF6vt507lqJ_qe-ST1PFE,2783
7
+ waterfall/watermark.py,sha256=avyQIFJBhqu_q_ZBp0-RWvAOIJmzJvVisbiIca2GPyA,14536
8
+ waterfall-0.2.10.dist-info/METADATA,sha256=vUr4PSIrQvPdkBjrsQc7uKJj4GVUSBZnpvElPA8n1Uc,8768
9
+ waterfall-0.2.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
10
+ waterfall-0.2.10.dist-info/entry_points.txt,sha256=XXnUzuWXu2nc9j4WAll9tq6HyodN_8WJLjeG0O4Y2Gw,60
11
+ waterfall-0.2.10.dist-info/licenses/LICENSE,sha256=zAtaO-k41Q-Q4Etl4bzuh7pgNJsPH-dYfzvznRa0OvM,11341
12
+ waterfall-0.2.10.dist-info/RECORD,,
waterfall-0.2.9.dist-info/RECORD DELETED
@@ -1,12 +0,0 @@
1
- waterfall/WatermarkerBase.py,sha256=ZyH1AWlXl1ak9lCnhtBdsd4IXsHKFGNnx_yYJFnf_zw,22018
2
- waterfall/WatermarkingFn.py,sha256=-b-kGRdL0a7eKRqJmcHPAR_rCjxQYnsg1Ne6bTwBc1I,1931
3
- waterfall/WatermarkingFnFourier.py,sha256=QYayAQYwi1dQkDIyqmvhU568VhrVYTVy47HkI8F8SZs,1358
4
- waterfall/WatermarkingFnSquare.py,sha256=2PAO05DdKT02npo7GDf_82D520nP7kGAWK6H4E4JMt4,1638
5
- waterfall/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
- waterfall/permute.py,sha256=uYKdmn4pGvjB6hInInLGxFIF6vt507lqJ_qe-ST1PFE,2783
7
- waterfall/watermark.py,sha256=avyQIFJBhqu_q_ZBp0-RWvAOIJmzJvVisbiIca2GPyA,14536
8
- waterfall-0.2.9.dist-info/METADATA,sha256=9sFs2kA_fj5CUD2UBt9uEuB1UTgn-ITJWUFWkksEqzE,8767
9
- waterfall-0.2.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
10
- waterfall-0.2.9.dist-info/entry_points.txt,sha256=XXnUzuWXu2nc9j4WAll9tq6HyodN_8WJLjeG0O4Y2Gw,60
11
- waterfall-0.2.9.dist-info/licenses/LICENSE,sha256=zAtaO-k41Q-Q4Etl4bzuh7pgNJsPH-dYfzvznRa0OvM,11341
12
- waterfall-0.2.9.dist-info/RECORD,,