ArissBandoss commited on
Commit
07fc3c2
·
verified ·
1 Parent(s): f0846b1

Update goai_helpers/goai_traduction.py

Browse files
Files changed (1) hide show
  1. goai_helpers/goai_traduction.py +1 -2
goai_helpers/goai_traduction.py CHANGED
@@ -43,9 +43,8 @@ def goai_traduction(text, src_lang, tgt_lang):
43
  forced_bos_token_id=tgt_lang_id,
44
  max_new_tokens=1024,
45
  num_beams=5,
46
- no_repeat_ngram_size=4,
47
  repetition_penalty=2.0,
48
- length_penalty=1.5,
49
  )
50
 
51
  # Décodage
 
43
  forced_bos_token_id=tgt_lang_id,
44
  max_new_tokens=1024,
45
  num_beams=5,
 
46
  repetition_penalty=2.0,
47
+ length_penalty=2,
48
  )
49
 
50
  # Décodage