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}")
     bad_words_ids = [[tokenizer.eos_token_id]]
     
-    for length_penalty in [1.0, 1.5, 2.0]:
+    for length_penalty in [1.0, 1.5, 2.0, 2.5, 3]:
         for num_beams in [5, 10]:
             print(f"\nTest avec length_penalty={length_penalty}, num_beams={num_beams}")