Commit 9af73872 authored by Tian Yun's avatar Tian Yun
Browse files

Fix max generation length

parent ad23a861
......@@ -375,7 +375,7 @@ class BaseLM(LM):
).to(self.device)
if max_generation_length is None:
max_length = self.max_gen_tok
max_length = self.max_gen_toks
else:
max_length = max_generation_length
......
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