ArissBandoss commited on
Commit
437a86e
·
verified ·
1 Parent(s): 242f488

Update goai_helpers/goai_traduction.py

Browse files
Files changed (1) hide show
  1. goai_helpers/goai_traduction.py +0 -4
goai_helpers/goai_traduction.py CHANGED
@@ -78,10 +78,6 @@ def translate_chunk(text, src_lang, tgt_lang):
78
  **inputs,
79
  forced_bos_token_id=tgt_lang_id,
80
  max_new_tokens=512,
81
- num_beams=5,
82
- no_repeat_ngram_size=4,
83
- repetition_penalty=2.0,
84
- length_penalty=1.0,
85
  early_stopping=True
86
  )
87
 
 
78
  **inputs,
79
  forced_bos_token_id=tgt_lang_id,
80
  max_new_tokens=512,
 
 
 
 
81
  early_stopping=True
82
  )
83