openbmb
/

cpm-bee-10b

Text Generation

feature-extraction

Model card Files Files and versions Community

Gong Baitao commited on Jun 16, 2023

Commit

b8e0001

•

1 Parent(s): 94ea18d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -45,8 +45,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from accelerate import dispatch_model
 from accelerate.utils import get_balanced_memory, infer_auto_device_map
-tokenizer = AutoTokenizer.from_pretrained("openbmb/cpm-bee-2b", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("openbmb/cpm-bee-2b", trust_remote_code=True).cuda()
 max_memory = get_balanced_memory(
     model,

 from accelerate import dispatch_model
 from accelerate.utils import get_balanced_memory, infer_auto_device_map
+tokenizer = AutoTokenizer.from_pretrained("openbmb/cpm-bee-10b", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("openbmb/cpm-bee-10b", trust_remote_code=True).cuda()
 max_memory = get_balanced_memory(
     model,