waterfall 0.2.4__py3-none-any.whl → 0.2.5__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 (6) hide show
  1. waterfall/WatermarkerBase.py +1 -1
  2. {waterfall-0.2.4.dist-info → waterfall-0.2.5.dist-info}/METADATA +1 -1
  3. {waterfall-0.2.4.dist-info → waterfall-0.2.5.dist-info}/RECORD +6 -6
  4. {waterfall-0.2.4.dist-info → waterfall-0.2.5.dist-info}/WHEEL +0 -0
  5. {waterfall-0.2.4.dist-info → waterfall-0.2.5.dist-info}/entry_points.txt +0 -0
  6. {waterfall-0.2.4.dist-info → waterfall-0.2.5.dist-info}/licenses/LICENSE +0 -0
waterfall/WatermarkerBase.py CHANGED
@@ -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.dist-info → waterfall-0.2.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: waterfall
3
- Version: 0.2.4
3
+ Version: 0.2.5
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.dist-info → waterfall-0.2.5.dist-info}/RECORD RENAMED
@@ -1,12 +1,12 @@
1
- waterfall/WatermarkerBase.py,sha256=nx2HhNCNy3yAhttq77cFnMmxlGucV_T8iVfbviAmClI,21046
1
+ waterfall/WatermarkerBase.py,sha256=WAngzoYKOZ9FMCP9jZGLChxpPMLYum05EsEbRsN59Bc,21080
2
2
  waterfall/WatermarkingFn.py,sha256=-b-kGRdL0a7eKRqJmcHPAR_rCjxQYnsg1Ne6bTwBc1I,1931
3
3
  waterfall/WatermarkingFnFourier.py,sha256=QYayAQYwi1dQkDIyqmvhU568VhrVYTVy47HkI8F8SZs,1358
4
4
  waterfall/WatermarkingFnSquare.py,sha256=2PAO05DdKT02npo7GDf_82D520nP7kGAWK6H4E4JMt4,1638
5
5
  waterfall/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
6
  waterfall/permute.py,sha256=uYKdmn4pGvjB6hInInLGxFIF6vt507lqJ_qe-ST1PFE,2783
7
7
  waterfall/watermark.py,sha256=IbH5r3oqjtKztDVryfDTr_NDn-CLZHow0S8nAEtZmdc,14420
8
- waterfall-0.2.4.dist-info/METADATA,sha256=wx9wThBNZFArRDHB3y21Kj0uZxtGV_qm1j2zKTjYEeU,8723
9
- waterfall-0.2.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
10
- waterfall-0.2.4.dist-info/entry_points.txt,sha256=XXnUzuWXu2nc9j4WAll9tq6HyodN_8WJLjeG0O4Y2Gw,60
11
- waterfall-0.2.4.dist-info/licenses/LICENSE,sha256=zAtaO-k41Q-Q4Etl4bzuh7pgNJsPH-dYfzvznRa0OvM,11341
12
- waterfall-0.2.4.dist-info/RECORD,,
8
+ waterfall-0.2.5.dist-info/METADATA,sha256=s5ZsmsLRM0f8JLovIuxD34JF6l2uu-JOvhFxG3O1ci4,8723
9
+ waterfall-0.2.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
10
+ waterfall-0.2.5.dist-info/entry_points.txt,sha256=XXnUzuWXu2nc9j4WAll9tq6HyodN_8WJLjeG0O4Y2Gw,60
11
+ waterfall-0.2.5.dist-info/licenses/LICENSE,sha256=zAtaO-k41Q-Q4Etl4bzuh7pgNJsPH-dYfzvznRa0OvM,11341
12
+ waterfall-0.2.5.dist-info/RECORD,,