Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adds score API to GPT-2 #1533

Merged
merged 2 commits into from
Mar 29, 2024
Merged

Conversation

RyanMullins
Copy link
Contributor

As with Gemma and Mistral, adding the .score() API to enable interpretability research with GPT-2.

@github-actions github-actions bot added the Gemma Gemma model specific issues label Mar 29, 2024
Copy link
Member

@mattdangerw mattdangerw left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm!

keras_nlp/models/gpt2/gpt2_causal_lm.py Outdated Show resolved Hide resolved
@mattdangerw mattdangerw added the kokoro:force-run Runs Tests on GPU label Mar 29, 2024
@kokoro-team kokoro-team removed the kokoro:force-run Runs Tests on GPU label Mar 29, 2024
@RyanMullins RyanMullins marked this pull request as ready for review March 29, 2024 18:06
@mattdangerw mattdangerw merged commit 3c25844 into keras-team:master Mar 29, 2024
6 checks passed
abuelnasr0 pushed a commit to abuelnasr0/keras-nlp that referenced this pull request Apr 2, 2024
* Adds score API to GPT-2

* Addressing reviewer comments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Gemma Gemma model specific issues
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants