TiberiuCristianLeon commited on
Commit
d945d59
·
verified ·
1 Parent(s): d91fa35

Update translator.py

Browse files
Files changed (1) hide show
  1. 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]