OpenBMB 开源社区/cpm-ant-10b
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

使用方法

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)