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

Add top_p logits processor and temperature rescaling to the multinomial sampler #646

Merged
merged 2 commits into from
Feb 12, 2024

Conversation

rlouf
Copy link
Member

@rlouf rlouf commented Feb 12, 2024

Closes #643. Closes #645

@rlouf rlouf added transformers Linked to the `transformers` integration samplers labels Feb 12, 2024
@rlouf rlouf merged commit 29bd1fe into outlines-dev:main Feb 12, 2024
5 checks passed
@rlouf rlouf deleted the add-top-p branch February 12, 2024 14:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
samplers transformers Linked to the `transformers` integration
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Allow to scale logits by temperature Add nucleus (top-p) sampling
1 participant