Update README.md
Browse files
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 |
|