Spaces:
Running
on
Zero
Running
on
Zero
Update goai_helpers/goai_traduction.py
Browse files
goai_helpers/goai_traduction.py
CHANGED
@@ -43,9 +43,8 @@ def goai_traduction(text, src_lang, tgt_lang):
|
|
43 |
forced_bos_token_id=tgt_lang_id,
|
44 |
max_new_tokens=1024,
|
45 |
num_beams=5,
|
46 |
-
no_repeat_ngram_size=4,
|
47 |
repetition_penalty=2.0,
|
48 |
-
length_penalty=
|
49 |
)
|
50 |
|
51 |
# Décodage
|
|
|
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
|