SnypzZz
/

Llama2-13b-Language-translate

text2text-generation

text-generation-inference

Model card Files Files and versions

SnypzZz commited on Jan 5, 2024

Commit

ea2e28e

·

1 Parent(s): 0b7d5cf

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -90,7 +90,7 @@ tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 # translate from English to Chinese
 generated_tokens = model.generate(
     **model_inputs,
-    forced_bos_token_id=tokenizer.lang_code_to_id["ko_KR"]
 )
 tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 # => '联合国首脑说,叙利亚没有军事解决办法'

 # translate from English to Chinese
 generated_tokens = model.generate(
     **model_inputs,
+    forced_bos_token_id=tokenizer.lang_code_to_id["zh_CN"]
 )
 tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 # => '联合国首脑说,叙利亚没有军事解决办法'