Update README.md

README.md

@@ -33,7 +33,7 @@ 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()
 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'
-model.inference_pretrain(prompt, tokenizer, model)
+model.inference_pretrain(prompt, tokenizer)
 ```
 
 ### 通过 Transformers 加载
@@ -44,7 +44,7 @@ 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()
 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'
-model.inference_pretrain(prompt, tokenizer, model)
+model.inference_pretrain(prompt, tokenizer)
 ```
 
 ### Open Source License