Reverts 6a35790

PiperOrigin-RevId: 720417398

Author: jetstream authors

google3/third_party/py/jetstream/engine/token_utils.py

@@ -335,7 +335,7 @@ def decode(self, token_ids: list[int], **kwargs) -> str:
     else:
       # If it's not streaming decoding, we can directly decode the full list
       # of token ids to a complete sequence.
-      return self.vocab.decode(token_ids)
+      return self.vocab.tokenizer.decode(token_ids)
 
   @property
   def pad_id(self) -> int: