Spaces:
Running
on
Zero
Running
on
Zero
Update goai_helpers/goai_traduction.py
Browse files
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
|