Fix model device in code example (#4)

- Fix model device in code example (02bda2ccbdc90faf06bebe95175e2a9f5f75be5c)


Co-authored-by: Lewis Tunstall <[email protected]>

README.md

@@ -225,7 +225,7 @@ checkpoint = "bigcode/santacoder"
 device = "cuda" # for GPU usage or "cpu" for CPU usage
 
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-model = AutoModelForCausalLM.from_pretrained(checkpoint, trust_remote_code=True).to()
+model = AutoModelForCausalLM.from_pretrained(checkpoint, trust_remote_code=True).to(device)
 
 inputs = tokenizer.encode("def print_hello_world():", return_tensors="pt").to(device)
 outputs = model.generate(inputs)