revert typehint

b3598058 · haileyschoelkopf · 5cb7af7e · b3598058
Commit b3598058 authored Jul 03, 2023 by haileyschoelkopf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/models/huggingface.py lm_eval/models/huggingface.py +1 -1

No files found.
--- a/lm_eval/models/huggingface.py
+++ b/lm_eval/models/huggingface.py
@@ -58,7 +58,7 @@ class HFLM(LM):

    def __init__(
        self,
-        device: Optional[Union[str, int]] = "cuda",
+        device="cuda",
        pretrained="gpt2",
        revision="main",
        low_cpu_mem_usage=None,