Skip to content

GitLab

Commit e33ae5ab authored by FarzanehNakhaee's avatar FarzanehNakhaee
Browse files

Update until in greedy_until

parent c08b2d9d
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
lm_eval/models/hf_causal.py
View file @ e33ae5ab
...@@ -372,6 +372,8 @@ class HFLM(LM): ...@@ -372,6 +372,8 @@ class HFLM(LM):
raise ValueError( raise ValueError(
f"Expected `gen_kwargs['until']` to be of type Union[str,list] but got {until}" f"Expected `gen_kwargs['until']` to be of type Union[str,list] but got {until}"
) )
else:
until = None
else: else:
raise ValueError( raise ValueError(
f"Expected `gen_kwargs` to be of type `dict` but got {gen_kwargs}" f"Expected `gen_kwargs` to be of type `dict` but got {gen_kwargs}"
......
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