Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -51,7 +51,7 @@ def goai_traduction(text, src_lang, tgt_lang):
     print(f"Token ID de la langue cible ({tgt_lang}): {tgt_lang_id}")
     
     for length_penalty in [1.0, 1.5, 2.0]:
-        for num_beams in [5, 8]:
+        for num_beams in [5, 10]:
             print(f"\nTest avec length_penalty={length_penalty}, num_beams={num_beams}")
             
 
@@ -62,6 +62,7 @@ def goai_traduction(text, src_lang, tgt_lang):
                 early_stopping=False,
                 num_beams=num_beams,
                 no_repeat_ngram_size=0,
+                bad_words_ids=bad_words_ids,
                 length_penalty=length_penalty
             )