Added GPT-Large model

brain-score · Nov 15, 2023 · 7ab3466 · 7ab3466
1 parent 7a8e0d1
commit 7ab3466
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/brainscore_language/models/gpt/__init__.py b/brainscore_language/models/gpt/__init__.py
@@ -11,6 +11,9 @@
 model_registry['gpt2-xl'] = lambda: HuggingfaceSubject(model_id='gpt2-xl', region_layer_mapping={
     ArtificialSubject.RecordingTarget.language_system: 'transformer.h.43'})
 
+model_registry['gpt2-large'] = lambda: HuggingfaceSubject(model_id='gpt2-large', region_layer_mapping={
+    ArtificialSubject.RecordingTarget.language_system: 'transformer.h.33'})
+
 model_registry['gpt-neo-2.7B'] = lambda: HuggingfaceSubject(model_id='EleutherAI/gpt-neo-2.7B', region_layer_mapping={
     ArtificialSubject.RecordingTarget.language_system: 'transformer.h.31'})