Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -90,7 +90,7 @@ tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
90
  # translate from English to Chinese
91
  generated_tokens = model.generate(
92
  **model_inputs,
93
- forced_bos_token_id=tokenizer.lang_code_to_id["zh_CN"]
94
  )
95
  tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
96
  # => '联合国首脑说,叙利亚没有军事解决办法'
 
90
  # translate from English to Chinese
91
  generated_tokens = model.generate(
92
  **model_inputs,
93
+ forced_bos_token_id=tokenizer.lang_code_to_id["ko_KR"]
94
  )
95
  tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
96
  # => '联合国首脑说,叙利亚没有军事解决办法'