Unverified Commit 59b047fb authored by Lintang Sutawika's avatar Lintang Sutawika Committed by GitHub
Browse files

Update task.py

parent 633fb458
......@@ -116,7 +116,9 @@ class TaskConfig(dict):
if self.output_type == "greedy_until":
# ensure that we greedily generate in absence of explicit arguments otherwise
self.generation_kwargs = {
"until": None if self.target_delimiter is None else [self.target_delimiter],
"until": None
if self.target_delimiter is None
else [self.target_delimiter],
"do_sample": False,
"temperature": 0.0,
}
......
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