wisent 0.5.11__py3-none-any.whl → 0.5.12__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.

Potentially problematic release.

This version of wisent might be problematic. Click here for more details.

Files changed (7) hide show
  1. wisent/__init__.py +1 -1
  2. wisent/core/models/wisent_model.py +6 -4
  3. {wisent-0.5.11.dist-info → wisent-0.5.12.dist-info}/METADATA +1 -1
  4. {wisent-0.5.11.dist-info → wisent-0.5.12.dist-info}/RECORD +7 -7
  5. {wisent-0.5.11.dist-info → wisent-0.5.12.dist-info}/WHEEL +0 -0
  6. {wisent-0.5.11.dist-info → wisent-0.5.12.dist-info}/licenses/LICENSE +0 -0
  7. {wisent-0.5.11.dist-info → wisent-0.5.12.dist-info}/top_level.txt +0 -0
wisent/__init__.py CHANGED
@@ -1 +1 @@
1
- __version__ = "0.5.11"
1
+ __version__ = "0.5.12"
wisent/core/models/wisent_model.py CHANGED
@@ -461,9 +461,10 @@ class WisentModel:
461
461
  truncation=True, # Avoid errors on long inputs
462
462
  max_length=self.tokenizer.model_max_length # Use model's actual limit
463
463
  )
464
+ # Move tensors to the correct device (same as _batch_encode does)
464
465
  batch = {
465
- "input_ids": tokenizer_output["input_ids"],
466
- "attention_mask": tokenizer_output["attention_mask"]
466
+ "input_ids": tokenizer_output["input_ids"].to(resolve_torch_device()),
467
+ "attention_mask": tokenizer_output["attention_mask"].to(resolve_torch_device())
467
468
  }
468
469
  else:
469
470
  # Current behavior: apply chat template
@@ -692,9 +693,10 @@ class WisentModel:
692
693
  truncation=True, # Avoid errors on long inputs
693
694
  max_length=self.tokenizer.model_max_length # Use model's actual limit
694
695
  )
696
+ # Move tensors to the correct device (same as _batch_encode does)
695
697
  batch = {
696
- "input_ids": tokenizer_output["input_ids"],
697
- "attention_mask": tokenizer_output["attention_mask"]
698
+ "input_ids": tokenizer_output["input_ids"].to(resolve_torch_device()),
699
+ "attention_mask": tokenizer_output["attention_mask"].to(resolve_torch_device())
698
700
  }
699
701
  else:
700
702
  # Current behavior: apply chat template
{wisent-0.5.11.dist-info → wisent-0.5.12.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: wisent
3
- Version: 0.5.11
3
+ Version: 0.5.12
4
4
  Summary: Monitor and guard against harmful content in language models
5
5
  Home-page: https://github.com/yourusername/wisent-activation-guardrails
6
6
  Author: Wisent Team
{wisent-0.5.11.dist-info → wisent-0.5.12.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- wisent/__init__.py,sha256=xFez9dUQrcuZqZRWuEIsCbMskoR-Ke1_uUZ51Kyt1tw,23
1
+ wisent/__init__.py,sha256=LznNzk7nDbJCv7NVxCOu958-1uT_nFJ79_3vJt7WPDc,23
2
2
  wisent/benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
3
  wisent/benchmarks/coding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
4
  wisent/benchmarks/coding/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -134,7 +134,7 @@ wisent/core/evaluators/oracles/interactive.py,sha256=f3v2_N17fKzGyeOxONRJbrbn8i5
134
134
  wisent/core/evaluators/oracles/nlp_evaluator.py,sha256=KxbnF-I2IFbBQpoYyjQKGbYh4NErsEuhTCRYX_Tob8o,18220
135
135
  wisent/core/evaluators/oracles/user_specified.py,sha256=V1dKrNj3Oq7UC_I7DT0WGnktP7R_DSW6UAwDdrA8SnE,2360
136
136
  wisent/core/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
137
- wisent/core/models/wisent_model.py,sha256=_XQpakCPJGdzeeSd0gPxp0yd057HisA69uAaW-katDo,31788
137
+ wisent/core/models/wisent_model.py,sha256=4W77IrPdVpgxjhnhybLuJpIFQmdc9e7TzmwxaRPRWVc,32052
138
138
  wisent/core/models/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
139
139
  wisent/core/models/core/atoms.py,sha256=_Bpz0Sfiq6_VswThIltUwNGj_ukl5MhAg8RrgMKwEBM,15756
140
140
  wisent/core/optuna/__init__.py,sha256=sTfwRnrRyKrCNVsF_qCjBDFEZC0ZmUZ7m6IE0iHfTVs,1914
@@ -213,8 +213,8 @@ wisent/synthetic/generators/diversities/core/__init__.py,sha256=47DEQpj8HBSa-_TI
213
213
  wisent/synthetic/generators/diversities/core/core.py,sha256=TjSj5T7NE5kRH-ABcFqb1Hz_j3Z6F_TcV-95uHD5Xw8,2201
214
214
  wisent/synthetic/generators/diversities/methods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
215
215
  wisent/synthetic/generators/diversities/methods/fast_diversity.py,sha256=Z2UzTbzyJFM_ToxCoXM_LQQQ1Jc6BZknrbpikTG1MRw,8522
216
- wisent-0.5.11.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
217
- wisent-0.5.11.dist-info/METADATA,sha256=na_FE2Pdt5j8nixhi8wFTOJYLzWweXPgylqL_3Pklx4,2425
218
- wisent-0.5.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
219
- wisent-0.5.11.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
220
- wisent-0.5.11.dist-info/RECORD,,
216
+ wisent-0.5.12.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
217
+ wisent-0.5.12.dist-info/METADATA,sha256=03T1uh1O8l6nuJQtIolh3yLGNBm-UPpwx0HNLZ2wntw,2425
218
+ wisent-0.5.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
219
+ wisent-0.5.12.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
220
+ wisent-0.5.12.dist-info/RECORD,,