Merge pull request #6 from robertsonwang/small_fixes

Small Fixes

src/synthid_text/logits_processing.py

@@ -270,7 +270,7 @@ def watermarked_call(
       ])
 
     device = scores.device
-    if device != self.device:
+    if str(device) != str(self.device):
       raise ValueError(
           "SynthIDLogitsProcessor received inputs with unexpected device.",
       )

src/synthid_text/logits_processing_test.py

@@ -93,7 +93,6 @@ def does_mean_g_value_matches_theoretical(
   )
 
   probs = torch.nn.functional.softmax(updated_scores, dim=1)
-  generator = torch.Generator(device=device).manual_seed(0)
   next_tokens = torch.multinomial(
       probs,
       num_samples=1,

src/synthid_text/synthid_mixin.py

@@ -210,7 +210,10 @@ def _sample(
           "`do_sample` is set to `True`, `logits_warper` must be a"
           f" `LogitsProcessorList` instance (it is {logits_warper})."
       )
-
+    if has_eos_stopping_criteria and not pad_token_id:
+      raise ValueError(
+          "`stopping_criteria` is not empty, `pad_token_id` must be set in generation_config."
+      )
     # init attention / hidden states / scores tuples
     scores = () if (return_dict_in_generate and output_scores) else None
     raw_logits = () if (return_dict_in_generate and output_logits) else None