openbmb
/

cpm-ant-10b

Text Generation

Model card Files Files and versions

jeffreygo commited on Jun 2, 2023

Commit

3b53c0f

·

1 Parent(s): 0c2f70d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -14,8 +14,8 @@ 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)

 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)