SnypzZz
/

Llama2-13b-Language-translate

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Update README.md

#5

by yeongseok11 - opened Jan 5

base: refs/heads/main

←

from: refs/pr/5

Discussion Files changed

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["zh_CN"]
 )
 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)
 # => '联合国首脑说,叙利亚没有军事解决办法'