Changed to many to one
parent
b4e95e23ba
commit
c1a0646630
|
|
@ -21,7 +21,7 @@ class MBartTranslator:
|
|||
tokenizer (MBart50TokenizerFast): The MBart tokenizer.
|
||||
"""
|
||||
|
||||
def __init__(self, model_name="facebook/mbart-large-50-many-to-many-mmt", src_lang=None, tgt_lang=None):
|
||||
def __init__(self, model_name="facebook/mbart-large-50-many-to-one-mmt", src_lang=None, tgt_lang=None):
|
||||
|
||||
self.supported_languages = [
|
||||
"ar_AR",
|
||||
|
|
@ -275,4 +275,4 @@ class Script(scripts.Script):
|
|||
|
||||
def postprocess(self, p, processed, *args):
|
||||
print(f"Post process: Translated prompt : {p.prompt}")
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue