Update README.md

README.md

@@ -337,12 +337,11 @@ We recommend using the latest version of the Transformers library (at least 4.47
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
- 
+import torch
 # 1. Load pre-trained model and tokenizer
-model_name = "baichuan-inc/Baichuan-M1-14B-Instruct"  
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
-
+model_name = "baichuan-inc/Baichuan-M1-14B-Base"  
+tokenizer = AutoTokenizer.from_pretrained(model_name,trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_name,trust_remote_code=True,torch_dtype = torch.bfloat16).cuda()
 # 2. Input prompt text
 prompt = "May I ask you some questions about medical knowledge?"