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)
- Downloads last month
- 8,112
Install from pip and serve model
# Install vLLM from pip: pip install vllm# Start the vLLM server: vllm serve "openbmb/cpm-ant-10b"# Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "openbmb/cpm-ant-10b", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'