Update README.md

README.md

@@ -27,7 +27,7 @@ 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_config(config, 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()