"llm/llama.cpp/.github/vscode:/vscode.git/clone" did not exist on "f00757db2784510506f52c629624ebe040183692"
Commit 732f7ed2 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

change generation_kwargs -> kwargs

parent 7ffb74ce
......@@ -511,11 +511,11 @@ class HFLM(LM):
until = [kwargs]
elif not isinstance(until, list):
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:
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:
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