Fix length related warnings in speculative decoding #29585
avoid generation length warning
7859c431
add tests
4b412499
gante
approved these changes
on 2024-03-11
Update src/transformers/generation/candidate_generator.py
7c35d05b
add tests and minor fixes
0cb8c1d2
refine `min_new_tokens`
f2b820ef
gante
commented
on 2024-03-12
Update src/transformers/generation/candidate_generator.py
dc8235c1
add method to prepare length arguments
2f482859
add test for min length
cc8a47cc
gante
commented
on 2024-03-13
Update src/transformers/generation/candidate_generator.py
0683f2d6
fix variable naming
24ed5d80
Merge remote-tracking branch 'upstream/main' into speculative_decoding
31ad6c02
Merge remote-tracking branch 'upstream/main' into speculative_decoding
2a2ec3d3
Merge remote-tracking branch 'upstream/main' into speculative_decoding
081e1b92
empty commit for tests
08ff6a9e
trigger tests (empty)
bb14b36f
Merge branch 'huggingface:main' into speculative_decoding
3b628e0d
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub