"sgl-kernel/vscode:/vscode.git/clone" did not exist on "f11481b921bb7f3eb7b1a9080667549326b7fd17"
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): ...@@ -116,7 +116,9 @@ class TaskConfig(dict):
if self.output_type == "greedy_until": if self.output_type == "greedy_until":
# ensure that we greedily generate in absence of explicit arguments otherwise # ensure that we greedily generate in absence of explicit arguments otherwise
self.generation_kwargs = { 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, "do_sample": False,
"temperature": 0.0, "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