CPM-Bee icon indicating copy to clipboard operation
CPM-Bee copied to clipboard

Where is /your/model/checkpoint in the code:model.load_state_dict(torch.load("/your/model/checkpoint"))?

Open backatonesJ opened this issue 1 year ago • 1 comments

I have download the model cpm-bee-10b from huggingface ,but I do not understand the path of /your/model/checkpointin. Can you someone please guide me to solve this problem?

backatonesJ avatar May 30 '23 07:05 backatonesJ

For example, you put the cpm-bee-10b model in path /root/user/cpm_bee_10b/pytorch_model.bin, the code will be model.load_state_dict(torch.load("/root/user/cpm_bee_10b/pytorch_model.bin"))

gongbaitao avatar May 30 '23 07:05 gongbaitao

thanks a lot the problem is solved

backatonesJ avatar Jun 01 '23 08:06 backatonesJ