Skip to content

Fix TransformerLens model name for GPT-2#17

Open
Tauheed-Elahee wants to merge 1 commit intoJoshEngels:mainfrom
Tauheed-Elahee:tl-model-name
Open

Fix TransformerLens model name for GPT-2#17
Tauheed-Elahee wants to merge 1 commit intoJoshEngels:mainfrom
Tauheed-Elahee:tl-model-name

Conversation

@Tauheed-Elahee
Copy link

TransformerLens expects "gpt2" not "gpt-2" as the model identifier. Add tl_model_name mapping so HookedTransformer.from_pretrained() receives the correct name.

TransformerLens expects "gpt2" not "gpt-2" as the model identifier.
Add tl_model_name mapping so HookedTransformer.from_pretrained()
receives the correct name.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant