Update README.md

README.md

@@ -78,7 +78,7 @@ prompt= """Below is an instruction that describes a task. Write a response that
 Choose three places you would like to visit and explain why.
 
 ### Response:"""
-inputs = tokenizer(prompt, return_tensors="pt", return_attention_mask=False)
+inputs = tokenizer(prompt, return_tensors="pt", return_attention_mask=False).to('cuda')
 outputs = model.generate(**inputs, max_length=500)
 text = tokenizer.batch_decode(outputs)[0]
 print(text)