rename tasks

This commit is contained in:
Xi Yan 2024-10-09 11:30:35 -07:00
parent 9c38d9ae13
commit ae43044a57
7 changed files with 5 additions and 6 deletions

View file

@ -44,7 +44,7 @@ async def run_main(host: str, port: int):
# CustomDataset
response = await client.run_evals(
"Llama3.2-1B-Instruct",
"Llama3.1-8B-Instruct",
"mmlu-simple-eval-en",
"mmlu",
)

View file

@ -90,7 +90,6 @@ class EleutherEvalsWrapper(LM):
def generate_until(self, requests, disable_tqdm: bool = False) -> List[str]:
res = []
for req in requests:
print("generation for msg: ", req.args[0])
response = self.inference_api.chat_completion(
model=self.model,
messages=[
@ -144,7 +143,7 @@ class EleutherEvalsAdapter(Evals):
output = evaluate(
eluther_wrapper,
task_dict,
limit=1,
limit=10,
)
formatted_output = lm_eval.utils.make_table(output)