Skip to content

GitLab

Commit b3598058 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

revert typehint

parent 5cb7af7e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/huggingface.py
View file @ b3598058
...@@ -58,7 +58,7 @@ class HFLM(LM): ...@@ -58,7 +58,7 @@ class HFLM(LM):
def __init__( def __init__(
self, self,
device: Optional[Union[str, int]] = "cuda", device="cuda",
pretrained="gpt2", pretrained="gpt2",
revision="main", revision="main",
low_cpu_mem_usage=None, low_cpu_mem_usage=None,
......
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