Unverified Commit 7a2ba052 authored by Surya Kasturi's avatar Surya Kasturi Committed by GitHub
Browse files

Capture gen_kwargs from CLI in squad_completion (#2727)



* Capture gen_kwargs from CLI in squad_completion

* Update lm_eval/tasks/squad_completion/task.py
Co-authored-by: default avatarBaber Abbasi <92168766+baberabb@users.noreply.github.com>

* Update lm_eval/tasks/squad_completion/task.py
Co-authored-by: default avatarBaber Abbasi <92168766+baberabb@users.noreply.github.com>

* pre-commit

---------
Co-authored-by: default avatarBaber Abbasi <92168766+baberabb@users.noreply.github.com>
Co-authored-by: default avatarBaber <baber@hey.com>
parent 7c9fbcf8
...@@ -2,6 +2,7 @@ import re ...@@ -2,6 +2,7 @@ import re
from typing import List from typing import List
import numpy as np import numpy as np
from deepcopy import deepcopy
from lm_eval.api.instance import Instance from lm_eval.api.instance import Instance
from lm_eval.api.task import ConfigurableTask from lm_eval.api.task import ConfigurableTask
...@@ -46,12 +47,14 @@ class SQUADCompletion(ConfigurableTask): ...@@ -46,12 +47,14 @@ class SQUADCompletion(ConfigurableTask):
language description, as well as the few shot examples, and the question language description, as well as the few shot examples, and the question
part of the document for `doc`. part of the document for `doc`.
""" """
arguments = deepcopy(self.config.generation_kwargs)
arguments["until"] = arguments.get("until", ["\n"])
arguments["max_gen_toks"] = arguments.get("max_gen_toks", 48)
return [ return [
Instance( Instance(
request_type="generate_until", request_type="generate_until",
doc=doc, doc=doc,
arguments=(ctx, {"until": ["\n"], "max_gen_toks": 48}), arguments=(ctx, arguments),
idx=0, idx=0,
**kwargs, **kwargs,
) )
......
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