Views
No views yet
pip install transformers1from transformers import CpmAntTokenizer, CpmAntForCausalLM
2
3texts = "今天天气不错,"
4model = CpmAntForCausalLM.from_pretrained("openbmb/cpm-ant-10b")
5tokenizer = CpmAntTokenizer.from_pretrained("openbmb/cpm-ant-10b")
6input_ids = tokenizer(texts, return_tensors="pt")
7outputs = model.generate(**input_ids)
8output_texts = tokenizer.batch_decode(outputs)
9
10print(output_texts)