metadata
tags:
- text-generation
language: zh
Usage
pip install transformers
from transformers import CpmAntTokenizer, CpmAntForCausalLM
texts = "今天天气不错，"
model = CpmAntForCausalLM.from_pretrained("openbmb/cpm-ant-10b")
tokenizer = CpmAntTokenizer.from_pretrained("openbmb/cpm-ant-10b")
input_ids = tokenizer(texts, return_tensors="pt")
outputs = model.generate(**input_ids)
output_texts = tokenizer.batch_decode(outputs)
print(output_texts)