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

Make LogitsProcessor compatible with torch.compile #29018

Closed
wants to merge 9 commits into from
Closed

Make LogitsProcessor compatible with torch.compile #29018

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
65cd4d2
make compatible with torch.compile
zucchini-nlp Feb 14, 2024
a240881
fixes and warnings
zucchini-nlp Feb 19, 2024
bfb5b18
run tests
zucchini-nlp Feb 20, 2024
fcc8437
Merge remote-tracking branch 'upstream/main' into feat/compile_generate
zucchini-nlp Feb 20, 2024
78ab061
make style
zucchini-nlp Feb 20, 2024
3750dd3
Update src/transformers/generation/logits_process.py
zucchini-nlp Feb 26, 2024
402bad6
Update src/transformers/generation/logits_process.py
zucchini-nlp Feb 26, 2024
f6d659c
Update src/transformers/generation/logits_process.py
zucchini-nlp Feb 26, 2024
35bd688
temporary changes, will not compile
zucchini-nlp Feb 26, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading