longvnhue1 commited on
Commit
0382d49
·
verified ·
1 Parent(s): 722170f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=1, # <-- Đơ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]}")
 
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]}")