Update README.md

README.md

@@ -39,9 +39,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 ckpt_path = "Mar2Ding/songcomposer_pretrain"
 tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
 model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
-from modeling_internlm2 import inference
-prompt = 'Create a song on brave and sacrificing with a rapid pace.'
-inference(prompt, tokenizer, model)
+from modeling_internlm2 import inference_pretrain
+prompt = '<bop> Total 7 lines. The first line:可,<D4>,<137>,<79>|惜,<D#4>,<137>,<79>|这,<F4>,<137>,<88>|是,<F4>,<121>,<79>|属,<F4>,<121>,<79>|于,<D#4>,<214>,<88>|你,<D#4>,<141>,<79>|的,<D4>,<130>,<79>|风,<C4>,<151>,<79>|景,<A#3> <F3>,<181><137>,<79>\n'
+inference_pretrain(prompt, tokenizer, model)
 ```
 
 ### 通过 Transformers 加载
@@ -52,9 +52,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 ckpt_path = "Mar2Ding/songcomposer_pretrain"
 tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
 model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
-from modeling_internlm2 import inference
-prompt = 'Create a song on brave and sacrificing with a rapid pace.'
-inference(prompt, tokenizer, model)
+from modeling_internlm2 import inference_pretrain
+prompt = '<bop> Total 7 lines. The first line:可,<D4>,<137>,<79>|惜,<D#4>,<137>,<79>|这,<F4>,<137>,<88>|是,<F4>,<121>,<79>|属,<F4>,<121>,<79>|于,<D#4>,<214>,<88>|你,<D#4>,<141>,<79>|的,<D4>,<130>,<79>|风,<C4>,<151>,<79>|景,<A#3> <F3>,<181><137>,<79>\n'
+inference_pretrain(prompt, tokenizer, model)
 ```
 
 ### Open Source License