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

Update goai_helpers/goai_traduction.py

Browse files
Files changed (1) hide show
  1. goai_helpers/goai_traduction.py +1 -4
goai_helpers/goai_traduction.py CHANGED
@@ -41,10 +41,7 @@ def goai_traduction(text, src_lang, tgt_lang):
41
  outputs = model.generate(
42
  **inputs,
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
 
41
  outputs = model.generate(
42
  **inputs,
43
  forced_bos_token_id=tgt_lang_id,
44
+ max_new_tokens=1024
 
 
 
45
  )
46
 
47
  # Décodage