Update README.md

README.md

@@ -60,7 +60,7 @@ template = open('templates/llama.txt').read()
 
 # Run the agent on a single state (text representation) and get the action
 state = template.format(**valid[0])
-agent = pipeline("McGill-NLP/Llama-3-8b-Web")
+agent = pipeline(model="McGill-NLP/Llama-3-8b-Web", device=0, torch_dtype='auto')
 out = agent(state, return_full_text=False)[0]
 print("Action:", out['generated_text'])