Remove the registration of "GPT2" as a model type

e131056a · Stella Biderman · GitHub · 79b972d6 · e131056a
Unverified Commit e131056a authored Jun 12, 2023 by Stella Biderman Committed by GitHub Jun 12, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/lm_eval/models/hf_causal.py
+++ b/lm_eval/models/hf_causal.py
@@ -15,7 +15,7 @@ from accelerate import Accelerator
 from itertools import islice


-@register_model("hf-causal", "gpt2")
+@register_model("hf-causal")
 class HFLM(LM):
    def __init__(
        self,