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.

Sign up for GitHub

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

Jump to bottom

Feature request: NoRepeatNgram #64

Open
cahya-wirawan opened this issue Feb 4, 2024 · 0 comments
Open

Feature request: NoRepeatNgram #64

cahya-wirawan opened this issue Feb 4, 2024 · 0 comments
Labels
backend enhancement New feature or request

Comments

@cahya-wirawan
Copy link
Contributor

Sometimes my model repeats a whole paragraph again and again endlessly. Would it be possible to ban repetition of specific ngram size, like this NoRepeatNgramLogitsProcessor from transformer https://huggingface.co/docs/transformers/v4.36.1/en/internal/generation_utils#transformers.NoRepeatNGramLogitsProcessor ?
Thanks.
@cryscan cryscan added enhancement New feature or request backend labels Feb 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
backend enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@cahya-wirawan @cryscan