Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -149,7 +149,7 @@ def translate_text(req: TranslateRequest):
|
|
149 |
**encoded,
|
150 |
forced_bos_token_id=tokenizer.get_lang_id(req.target_lang),
|
151 |
max_length=256, # <-- Giữ mức vừa phải
|
152 |
-
num_beams=
|
153 |
no_repeat_ngram_size=3,
|
154 |
)
|
155 |
print(f"forced_bos_token_id: {tokenizer.lang_code_to_id[req.target_lang]}")
|
|
|
149 |
**encoded,
|
150 |
forced_bos_token_id=tokenizer.get_lang_id(req.target_lang),
|
151 |
max_length=256, # <-- Giữ mức vừa phải
|
152 |
+
num_beams=2, # <-- Đơn giản beam search
|
153 |
no_repeat_ngram_size=3,
|
154 |
)
|
155 |
print(f"forced_bos_token_id: {tokenizer.lang_code_to_id[req.target_lang]}")
|