waterfall 0.2.4__tar.gz → 0.2.6__tar.gz

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 (24) hide show
  1. {waterfall-0.2.4 → waterfall-0.2.6}/.gitignore +1 -2
  2. {waterfall-0.2.4 → waterfall-0.2.6}/PKG-INFO +1 -1
  3. {waterfall-0.2.4 → waterfall-0.2.6}/pyproject.toml +1 -1
  4. {waterfall-0.2.4 → waterfall-0.2.6}/waterfall/WatermarkerBase.py +1 -1
  5. waterfall-0.2.4/.vscode/settings.json +0 -5
  6. waterfall-0.2.4/decode.gif +0 -0
  7. waterfall-0.2.4/demo.ipynb +0 -3530
  8. waterfall-0.2.4/dropper.png +0 -0
  9. waterfall-0.2.4/test_animation.py +0 -60
  10. waterfall-0.2.4/test_vllm.ipynb +0 -697
  11. waterfall-0.2.4/token_tree_generator.py +0 -501
  12. {waterfall-0.2.4 → waterfall-0.2.6}/Images/Illustration.gif +0 -0
  13. {waterfall-0.2.4 → waterfall-0.2.6}/Images/Problem_formulation.jpg +0 -0
  14. {waterfall-0.2.4 → waterfall-0.2.6}/Images/Watermarking_process.png +0 -0
  15. {waterfall-0.2.4 → waterfall-0.2.6}/LICENSE +0 -0
  16. {waterfall-0.2.4 → waterfall-0.2.6}/README.md +0 -0
  17. {waterfall-0.2.4 → waterfall-0.2.6}/__init__.py +0 -0
  18. {waterfall-0.2.4 → waterfall-0.2.6}/requirements.txt +0 -0
  19. {waterfall-0.2.4 → waterfall-0.2.6}/waterfall/WatermarkingFn.py +0 -0
  20. {waterfall-0.2.4 → waterfall-0.2.6}/waterfall/WatermarkingFnFourier.py +0 -0
  21. {waterfall-0.2.4 → waterfall-0.2.6}/waterfall/WatermarkingFnSquare.py +0 -0
  22. {waterfall-0.2.4 → waterfall-0.2.6}/waterfall/__init__.py +0 -0
  23. {waterfall-0.2.4 → waterfall-0.2.6}/waterfall/permute.py +0 -0
  24. {waterfall-0.2.4 → waterfall-0.2.6}/waterfall/watermark.py +0 -0
{waterfall-0.2.4 → waterfall-0.2.6}/.gitignore RENAMED
@@ -3,5 +3,4 @@
3
3
  *.egg-info
4
4
  /.python-version
5
5
  /.idea
6
- dist
7
- media
6
+ dist
{waterfall-0.2.4 → waterfall-0.2.6}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: waterfall
3
- Version: 0.2.4
3
+ Version: 0.2.6
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
{waterfall-0.2.4 → waterfall-0.2.6}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
4
4
 
5
5
  [project]
6
6
  name = "waterfall"
7
- version = "0.2.4"
7
+ version = "0.2.6"
8
8
  authors = [
9
9
  { name = "Xinyuan Niu", email="aperture@outlook.sg" }
10
10
  ]
{waterfall-0.2.4 → waterfall-0.2.6}/waterfall/WatermarkerBase.py RENAMED
@@ -116,7 +116,7 @@ class Watermarker:
116
116
  self.N = self.tokenizer.vocab_size
117
117
 
118
118
  if isinstance(model, str):
119
- self.load_model(model, device_map=device, dtype=dtype)
119
+ self.load_model(model, device_map=device if device is not None else "auto", dtype=dtype)
120
120
  else:
121
121
  self.model = model
122
122
 
waterfall-0.2.4/.vscode/settings.json DELETED
@@ -1,5 +0,0 @@
1
- {
2
- "python-envs.defaultEnvManager": "ms-python.python:conda",
3
- "python-envs.defaultPackageManager": "ms-python.python:conda",
4
- "python-envs.pythonProjects": []
5
- }
waterfall-0.2.4/decode.gif DELETED
Binary file