NiuTrans
/

LMT-60-8B

text-generation

text-generation-inference

Model card Files Files and versions

luoyingfeng commited on 19 days ago

Commit

c73e0a6

·

verified ·

1 Parent(s): 22fc475

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -101,9 +101,9 @@ model_name = "NiuTrans/LMT-60-8B"
 tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side='left')
 model = AutoModelForCausalLM.from_pretrained(model_name)
-prompt = "Translate the following text from English into Chinese.
 English: The concept came from China where plum blossoms were the flower of choice.
-Chinese:"
 messages = [{"role": "user", "content": prompt}]
 text = tokenizer.apply_chat_template(
     messages,

 tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side='left')
 model = AutoModelForCausalLM.from_pretrained(model_name)
+prompt = """Translate the following text from English into Chinese.
 English: The concept came from China where plum blossoms were the flower of choice.
+Chinese:"""
 messages = [{"role": "user", "content": prompt}]
 text = tokenizer.apply_chat_template(
     messages,