diff --git a/brainscore_language/models/gpt/__init__.py b/brainscore_language/models/gpt/__init__.py index f75d9d10..0f91881c 100644 --- 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'})