Update README.md

README.md

@@ -46,7 +46,10 @@ Here provides a code snippet with `apply_chat_template` to show you how to load
 from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # the device to load the model onto
 
-model = AutoModelForCausalLM.from_pretrained("Qwen2/Qwen2-beta-4B-Chat", device_map="auto")
+model = AutoModelForCausalLM.from_pretrained(
+    "Qwen2/Qwen2-beta-4B-Chat",
+    device_map="auto"
+)
 tokenizer = AutoTokenizer.from_pretrained("Qwen2/Qwen2-beta-4B-Chat")
 
 prompt = "Give me a short introduction to large language model."
@@ -55,7 +58,8 @@ messages = [
     {"role": "user", "content": prompt}
 ]
 text = tokenizer.apply_chat_template(
-    messages, tokenize=False,
+    messages,
+    tokenize=False,
     add_generation_prompt=True
 )
 model_inputs = tokenizer([text], return_tensors="pt").to(device)