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