Skip to content

GitLab

Unverified Commit 6ae376e3 authored by Lintang Sutawika's avatar Lintang Sutawika Committed by GitHub
Browse files

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

Remove the registration of "GPT2" as a model type
parents 6db83ddc e131056a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/hf_causal.py
View file @ 6ae376e3
...@@ -15,7 +15,7 @@ from accelerate import Accelerator ...@@ -15,7 +15,7 @@ from accelerate import Accelerator
from itertools import islice from itertools import islice
@register_model("hf-causal", "gpt2") @register_model("hf-causal")
class HFLM(LM): class HFLM(LM):
def __init__( def __init__(
self, self,
......
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