Bo1015
/

proteinglm-3b-clm

Inference Endpoints

Model card Files Files and versions Community

Bo1015 commited on Jul 14

Commit

8504a58

•

1 Parent(s): 1674a28

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -26,8 +26,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig
 import torch
 tokenizer  = AutoTokenizer.from_pretrained("biomap-research/xtrimopglm-3b-clm", trust_remote_code=True, use_fast=True)
-config = AutoConfig.from_pretrained("biomap-research/xtrimopglm-3b-clm",  trust_remote_code=True, torch_dtype=torch.bfloat16)
-model = AutoModelForCausalLM.from_pretrained(config, trust_remote_code=True, torch_dtype=torch.bfloat16)
 if torch.cuda.is_available():
     model = model.cuda()
 model.eval()

 import torch
 tokenizer  = AutoTokenizer.from_pretrained("biomap-research/xtrimopglm-3b-clm", trust_remote_code=True, use_fast=True)
+model = AutoModelForCausalLM.from_pretrained("biomap-research/xtrimopglm-3b-clm", trust_remote_code=True, torch_dtype=torch.bfloat16)
 if torch.cuda.is_available():
     model = model.cuda()
 model.eval()