waterfall 0.1.2__py3-none-any.whl → 0.1.3__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.
@@ -1,23 +1,22 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: waterfall
3
- Version: 0.1.2
3
+ Version: 0.1.3
4
4
  Summary: Scalable Framework for Robust Text Watermarking and Provenance for LLMs
5
- Author-email: Xinyuan Niu <aperture@outlook.sg>
6
- License-Expression: Apache-2.0
7
5
  Project-URL: Homepage, https://github.com/aoi3142/Waterfall
8
6
  Project-URL: Issues, https://github.com/aoi3142/Waterfall/issues
9
- Classifier: Programming Language :: Python :: 3
7
+ Author-email: Xinyuan Niu <aperture@outlook.sg>
8
+ License-Expression: Apache-2.0
9
+ License-File: LICENSE
10
10
  Classifier: Operating System :: OS Independent
11
+ Classifier: Programming Language :: Python :: 3
11
12
  Requires-Python: >=3.10
12
- Description-Content-Type: text/markdown
13
- License-File: LICENSE
14
13
  Requires-Dist: accelerate>=0.29.0
15
14
  Requires-Dist: numpy>=2.0.0
16
15
  Requires-Dist: scipy>=1.13.0
17
16
  Requires-Dist: sentence-transformers>=3.0.0
18
17
  Requires-Dist: torch>=2.3.0
19
18
  Requires-Dist: transformers>=4.43.1
20
- Dynamic: license-file
19
+ Description-Content-Type: text/markdown
21
20
 
22
21
  # Waterfall: Scalable Framework for Robust Text Watermarking and Provenance for LLMs [EMNLP 2024 Main Long]
23
22
  Gregory Kang Ruey Lau*, Xinyuan Niu*, Hieu Dao, Jiangwei Chen, Chuan-Sheng Foo, Bryan Kian Hsiang Low
@@ -53,7 +52,7 @@ Protecting intellectual property (IP) of text such as articles and code is incre
53
52
 
54
53
  5. A token is sampled from the perturbed logits $\check{L}$ and is appended to the watermarked text.
55
54
 
56
- 6. Append the generated token to the prompt and continue autoregressive generation (steps 1-5) until the eos token.
55
+ 6. Append the generated token to the prompt and continue autoregressive generation (steps 1-5) until the `eos` token.
57
56
 
58
57
  # Verification of un/watermarked text
59
58
 
@@ -77,12 +76,12 @@ Protecting intellectual property (IP) of text such as articles and code is incre
77
76
 
78
77
  # Using our code
79
78
 
80
- Install our package from pypi
79
+ Install our package using `pip`
81
80
  ```sh
82
81
  pip install waterfall
83
82
  ```
84
83
 
85
- ## Alterntive installation from source
84
+ ## Alternative installation from source
86
85
 
87
86
  [Optional]
88
87
  If using `conda` (or other pkg managers), it is highly advisable to create a new environment
@@ -92,7 +91,7 @@ conda create -n waterfall python=3.11 --yes `# Compatible with python version
92
91
  conda activate waterfall
93
92
  ```
94
93
 
95
- Clone and install our package from source
94
+ Clone and install our package
96
95
  ```sh
97
96
  git clone https://github.com/aoi3142/Waterfall.git
98
97
  pip install -e Waterfall `# Install in 'editable' mode with '-e', can be omitted`
@@ -105,7 +104,7 @@ Use the command `waterfall_demo` to watermark a piece of text, and then verify t
105
104
  waterfall_demo
106
105
  ```
107
106
 
108
- \* Ensure that your device (cuda/cpu/mps) has enough memory to load the model (~16GB for default Llama 3.1 8B model)
107
+ \* Ensure that your device (`cuda`/`cpu`/`mps`) has enough memory to load the model and perform inference (~18GB+ for default Llama 3.1 8B model)
109
108
 
110
109
  Additional arguments
111
110
  ```sh
@@ -5,9 +5,8 @@ waterfall/WatermarkingFnSquare.py,sha256=2PAO05DdKT02npo7GDf_82D520nP7kGAWK6H4E4
5
5
  waterfall/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
6
  waterfall/permute.py,sha256=RwxOHFhx_VSOhhFwy5s79YgwTUBkfW2-LCCXYR3VT2o,2582
7
7
  waterfall/watermark.py,sha256=whiNhPwWNNIZwXMH6r7QzEE3A7Niq2Ro9elA1iSRoxI,11952
8
- waterfall-0.1.2.dist-info/licenses/LICENSE,sha256=zAtaO-k41Q-Q4Etl4bzuh7pgNJsPH-dYfzvznRa0OvM,11341
9
- waterfall-0.1.2.dist-info/METADATA,sha256=zDciBMgLwmhhf8voqFzLeCOruNQBdqtohMUYKDMzmdQ,8715
10
- waterfall-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
11
- waterfall-0.1.2.dist-info/entry_points.txt,sha256=XXnUzuWXu2nc9j4WAll9tq6HyodN_8WJLjeG0O4Y2Gw,60
12
- waterfall-0.1.2.dist-info/top_level.txt,sha256=5rTgijeT9V5GRCwIDZmhjeZ4khgH1lmfhS9ZmdUUCKQ,10
13
- waterfall-0.1.2.dist-info/RECORD,,
8
+ waterfall-0.1.3.dist-info/METADATA,sha256=hnX9Vq2zjuq_u9VxJcP9OQ5U2RHUtTGnTQu6iwUJhXM,8715
9
+ waterfall-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
10
+ waterfall-0.1.3.dist-info/entry_points.txt,sha256=XXnUzuWXu2nc9j4WAll9tq6HyodN_8WJLjeG0O4Y2Gw,60
11
+ waterfall-0.1.3.dist-info/licenses/LICENSE,sha256=zAtaO-k41Q-Q4Etl4bzuh7pgNJsPH-dYfzvznRa0OvM,11341
12
+ waterfall-0.1.3.dist-info/RECORD,,
@@ -1,5 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.9.0)
2
+ Generator: hatchling 1.27.0
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
-
@@ -1 +0,0 @@
1
- waterfall