xhluca commited on
Commit
09856c2
1 Parent(s): dcf6ae5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -60,7 +60,7 @@ template = open('templates/llama.txt').read()
60
 
61
  # Run the agent on a single state (text representation) and get the action
62
  state = template.format(**valid[0])
63
- agent = pipeline("McGill-NLP/Llama-3-8b-Web")
64
  out = agent(state, return_full_text=False)[0]
65
  print("Action:", out['generated_text'])
66
 
 
60
 
61
  # Run the agent on a single state (text representation) and get the action
62
  state = template.format(**valid[0])
63
+ agent = pipeline(model="McGill-NLP/Llama-3-8b-Web", device=0, torch_dtype='auto')
64
  out = agent(state, return_full_text=False)[0]
65
  print("Action:", out['generated_text'])
66