Changed to many to one

pull/10/head
Saifeddine ALOUI 2023-03-28 13:12:09 +02:00 committed by GitHub
parent b4e95e23ba
commit c1a0646630
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -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}")