Update README.md

README.md

@@ -126,6 +126,9 @@ We release DiscoLeo-8B in the following configurations:
 Here's how to use the model with transformers:
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+device="cuda"
 
 model = AutoModelForCausalLM.from_pretrained(
     "DiscoResearch/Llama3-DiscoLeo-Instruct-8B-v0.1",