diff --git a/README.md b/README.md
index 76b1a7e..827d5c4 100644
--- a/README.md
+++ b/README.md
@@ -46,7 +46,7 @@ You are trained by OpenBuddy team, (https://openbuddy.ai, https://github.com/Ope
 
 User: 睡不着怎么办?
 Assistant:"""
-inputs = tokenizer.encode(prompt, return_tensors="pt")
+inputs = tokenizer.encode(prompt, return_tensors="pt").cuda()
 outputs = model.generate(inputs, max_length=512)
 response = tokenizer.decode(outputs[0])
 print(response)