Update README.md

README.md

@@ -83,8 +83,8 @@ trainset = Dataset()  # Make sure trainset.__getitem__() can get data with corre
 # for details, you can read https://github.com/OpenBMB/CPM-Bee/tree/main/tutorials/basic_task_finetune
 train_loader = DataLoader(trainset, batch_size=1)
 
-tokenizer = AutoTokenizer.from_pretrained("openbmb/cpm-bee-1b", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("openbmb/cpm-bee-1b", trust_remote_code=True).cuda()
+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()
 
 optimizer = torch.optim.Adam(model.parameters())