Update README.md

README.md

@@ -33,7 +33,7 @@ tokenizer = AutoTokenizer.from_pretrained(
     )
 tokenizer.pad_token = tokenizer.eos_token
 text = "This is text."
-inputs = tokenizer(text, return_tensors="pt", padding=True)
+inputs = tokenizer(text, return_tensors="pt", padding=True).to('cuda')
 ```
 
 To modify the latent feature $h_0$ (`SCAR.hook.mod_features = 0`) of the SAE do the following: