Update translator.py
Browse files- translator.py +1 -1
translator.py
CHANGED
|
@@ -409,7 +409,7 @@ class Translators:
|
|
| 409 |
def mbart_one_to_many(self):
|
| 410 |
# translate from English
|
| 411 |
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
| 412 |
-
model = MBartForConditionalGeneration.from_pretrained(self.model_name, device_map="auto", torch_dtype="auto").to(self.device)
|
| 413 |
tokenizer = MBart50TokenizerFast.from_pretrained(self.model_name, src_lang="en_XX")
|
| 414 |
model_inputs = tokenizer(self.input_text, return_tensors="pt")
|
| 415 |
langid = languagecodes.mbart_large_languages[self.tl]
|
|
|
|
| 409 |
def mbart_one_to_many(self):
|
| 410 |
# translate from English
|
| 411 |
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
| 412 |
+
model = MBartForConditionalGeneration.from_pretrained(self.model_name, device_map="auto", torch_dtype="auto").to(self.device)
|
| 413 |
tokenizer = MBart50TokenizerFast.from_pretrained(self.model_name, src_lang="en_XX")
|
| 414 |
model_inputs = tokenizer(self.input_text, return_tensors="pt")
|
| 415 |
langid = languagecodes.mbart_large_languages[self.tl]
|