AriNubar commited on
Commit
2f130e3
·
verified ·
1 Parent(s): 6188c7d

Update translation.py

Browse files
Files changed (1) hide show
  1. translation.py +1 -1
translation.py CHANGED
@@ -135,7 +135,7 @@ class Translator:
135
  ):
136
  self.tokenizer.src_lang = src_lang
137
  encoded = self.tokenizer(
138
- text, return_tensors="pt", truncation=True, max_length=256
139
  )
140
  if max_length == "auto":
141
  max_length = int(32 + 2.0 * encoded.input_ids.shape[1])
 
135
  ):
136
  self.tokenizer.src_lang = src_lang
137
  encoded = self.tokenizer(
138
+ text, return_tensors="pt", truncation=True, max_length=256, padding=True,
139
  )
140
  if max_length == "auto":
141
  max_length = int(32 + 2.0 * encoded.input_ids.shape[1])