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): ...@@ -104,8 +104,9 @@ class TaskConfig(dict):
if self.generation_kwargs is not None: if self.generation_kwargs is not None:
if self.output_type != "greedy_until": if self.output_type != "greedy_until":
eval_logger.warning( 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: if "temperature" in self.generation_kwargs:
self.generation_kwargs["temperature"] = float( 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