Update README.md

README.md

@@ -54,9 +54,9 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM  
 
 model_name = 'davidsi/Llama3_1-8B-Instruct-AMD-python'  
-tokenizer = AutoTokenizer.from_pretrained(model_name)  
-llm = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16).to(device)
 
 messages = [
     {"role": "system", "content": "You are a helpful assistant for AMD technologies and python."},