Update README.md

README.md

@@ -9,14 +9,15 @@ base_model:
 The CultureSPA model from https://github.com/shaoyangxu/CultureSPA
 
 ```
-# load
+## load
 device = torch.device("cuda")
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(
     model_path,
     torch_dtype=torch.bfloat16
-)
-# inference
+).to(device)
+
+## inference
 messages = [
     {"role": "system", "content": system_prompt},
     {"role": "user", "content": instruction},