Merge pull request #574 from EleutherAI/StellaAthena-patch-1

Remove the registration of "GPT2" as a model type

Merge pull request #574 from EleutherAI/StellaAthena-patch-1
Remove the registration of "GPT2" as a model type
6ae376e3 · Lintang Sutawika · GitHub · 6db83ddc · e131056a · 6ae376e3
Unverified Commit 6ae376e3 authored Jun 12, 2023 by Lintang Sutawika 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,