Commit 7dd1ef95 authored by jon-tow's avatar jon-tow
Browse files

Fix naming

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