Update goai_helpers/goai_traduction.py
Browse files
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 |
|