Moore-Language-Space-ZeroGPU

Running on Zero

ArissBandoss commited on 3 days ago

Commit

4baecca

verified ·

1 Parent(s): f87780e

Update goai_helpers/goai_traduction.py

Files changed (1) hide show

goai_helpers/goai_traduction.py CHANGED Viewed

@@ -60,7 +60,7 @@ def translate_chunk(text, src_lang, tgt_lang):
         model_id = "ArissBandoss/mos2fr-3B"
     else:
         #model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"
-        model_id = "ArissBandoss/nllb-200-3.3B-fr2mos-peft"
     tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)

         model_id = "ArissBandoss/mos2fr-3B"
     else:
         #model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"
+        model_id = "ArissBandoss/nllb-200-3.3B-mos-fr-bidirectional-peft"
     tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)