Commit 80358e5a authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

add assert for passing generation_kwargs w/ output_type neq greedy_until

parent fc63c7af
......@@ -104,8 +104,9 @@ class TaskConfig(dict):
if self.generation_kwargs is not None:
if self.output_type != "greedy_until":
eval_logger.warning(
"passed `generation_kwargs`, but not using a generation request type!"
"passed `generation_kwargs`, but not using `output_type: greedy_until`!"
)
assert self.output_type != "greedy_until"
if "temperature" in self.generation_kwargs:
self.generation_kwargs["temperature"] = float(
......
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