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

Conform output of TextGenerationPipelines with HuggingfacePipelines #1205

Merged
merged 2 commits into from
Aug 25, 2023

Conversation

rahul-tuli
Copy link
Member

Our pipelines right now generate one extra token per inference than hugging face pipelines, this PR fixes that.

Additionally this also matches the generated logits with generated tokens.

@rahul-tuli rahul-tuli marked this pull request as ready for review August 25, 2023 18:03
@rahul-tuli rahul-tuli merged commit 31a71a9 into main Aug 25, 2023
7 checks passed
@rahul-tuli rahul-tuli deleted the conform-with-huggingface branch August 25, 2023 19:20
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.

None yet

5 participants