ArissBandoss commited on
Commit
d79365c
·
verified ·
1 Parent(s): d20c6c4

Update goai_helpers/goai_traduction.py

Browse files
Files changed (1) hide show
  1. goai_helpers/goai_traduction.py +1 -0
goai_helpers/goai_traduction.py CHANGED
@@ -40,6 +40,7 @@ def goai_traduction(text, src_lang, tgt_lang):
40
  forced_bos_token_id=tgt_lang_id,
41
  max_new_tokens=1024,
42
  early_stopping=False,
 
43
  no_repeat_ngram_size=0,
44
  length_penalty=1.0
45
  )
 
40
  forced_bos_token_id=tgt_lang_id,
41
  max_new_tokens=1024,
42
  early_stopping=False,
43
+ num_beams=5,
44
  no_repeat_ngram_size=0,
45
  length_penalty=1.0
46
  )