Skip to content

GitLab

Commit 2e27053d authored by daniel-furman's avatar daniel-furman
Browse files

llama test

parent 1ea8470c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/huggingface.py
View file @ 2e27053d
...@@ -697,7 +697,7 @@ class HFLM(LM): ...@@ -697,7 +697,7 @@ class HFLM(LM):
for element in elements[1:-1]: for element in elements[1:-1]:
new_elements.append(element.strip()) new_elements.append(element.strip())
new_elements new_elements
chat = [] chat = [{"role": "system", "content": f"You are a helpful and concise assistant."}]
for i in range(len(new_elements)): for i in range(len(new_elements)):
if i % 2 == 0: if i % 2 == 0:
chat.append({"role": "user", "content": f"Question: {new_elements[i]} Answer:"}) chat.append({"role": "user", "content": f"Question: {new_elements[i]} Answer:"})
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment