Spaces:
Running
Running
Update translation.py
Browse files- 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])
|