Commit 732f7ed2 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

change generation_kwargs -> kwargs

parent 7ffb74ce
...@@ -511,11 +511,11 @@ class HFLM(LM): ...@@ -511,11 +511,11 @@ class HFLM(LM):
until = [kwargs] until = [kwargs]
elif not isinstance(until, list): elif not isinstance(until, list):
raise ValueError( raise ValueError(
f"Expected `generation_kwargs['until']` to be of type Union[str,list] but got {until}" f"Expected `kwargs['until']` to be of type Union[str,list] but got {until}"
) )
else: else:
raise ValueError( raise ValueError(
f"Expected `generation_kwargs` to be of type `dict` but got {kwargs}" f"Expected `kwargs` to be of type `dict` but got {kwargs}"
) )
if not until: if not until:
until = [self.tok_decode(self.eot_token_id)] until = [self.tok_decode(self.eot_token_id)]
......
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