Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -39,9 +39,13 @@ def goai_traduction(text, src_lang, tgt_lang):
     
     # Bloquer complètement le token EOS jusqu'à un certain point
     outputs = model.generate(
-        **inputs,
-        forced_bos_token_id=tgt_lang_id,
-        max_new_tokens=1024
+         **inputs,
+         forced_bos_token_id=tgt_lang_id,
+         max_new_tokens=1024,
+         early_stopping=False,
+         num_beams=5,
+         no_repeat_ngram_size=0,
+         length_penalty=1.0
     )
     
     # Décodage