Views
No views yet
1from transformers import AutoTokenizer, AutoModel
2ckpt_path = "Mar2Ding/songcomposer_pretrain"
3tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
4model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
5prompt = '<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'
6model.inference_pretrain(prompt, tokenizer)1from transformers import AutoTokenizer, AutoModel
2ckpt_path = "Mar2Ding/songcomposer_pretrain"
3tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
4model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
5prompt = '<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'
6model.inference_pretrain(prompt, tokenizer)