correct position of decode args

michaelfeil · May 20, 2023 · e236f00 · e236f00
1 parent 3ed2b81
commit e236f00
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/hf_hub_ctranslate2/translate.py b/hf_hub_ctranslate2/translate.py
@@ -117,7 +117,8 @@ def _forward(self, *args, **kwds):
     def tokenize_decode(self, tokens_out, *args, **kwargs):
         return [
             self.tokenizer.decode(
-                self.tokenizer.convert_tokens_to_ids(tokens_out[i].hypotheses[0], *args, **kwargs)
+                self.tokenizer.convert_tokens_to_ids(tokens_out[i].hypotheses[0]),
+                *args, **kwargs
             )
             for i in range(len(tokens_out))
         ]
@@ -210,7 +211,8 @@ def tokenize_encode(self, text, *args, **kwargs):
     def tokenize_decode(self, tokens_out, *args, **kwargs):
         return [
             self.tokenizer.decode(
-                self.tokenizer.convert_tokens_to_ids(tokens_out[i].hypotheses[0][1:], *args, **kwargs)
+                self.tokenizer.convert_tokens_to_ids(tokens_out[i].hypotheses[0][1:]),
+                *args, **kwargs
             )
             for i in range(len(tokens_out))
         ]