Undo another change

Author: riedgar-ms

guidance/models/transformers/_transformers.py

@@ -121,10 +121,11 @@ def _tokenizer(self, model, **kwargs):
         ), "You must give a model name when you provide a tokenizer object!"
 
         return tokenizer
-    
+
     def __call__(self, byte_string):
         tokenisation = self._orig_tokenizer(byte_string)
-        return tokenisation['input_ids']
+        return tokenisation["input_ids"]
+
 
 class TransformersEngine(Engine):
     def __init__(self, model, tokenizer, compute_log_probs, **kwargs):
@@ -271,7 +272,7 @@ def get_logits(self, token_ids, forced_bytes, current_temp):
             self.metrics.model_input_tokens += len(new_token_ids)
             self.metrics.model_output_tokens += 1
 
-        return self._cached_logits, len(new_token_ids)
+        return self._cached_logits
 
 
 class Transformers(Model):