PR #28759 fix num_assistant_tokens with heuristic schedule

fix num_assistant_tokens with heuristic schedule #28759

amyeroberts merged 17 commits into huggingface:main from jmamou:fix_num_assistant_tokens

fix heuristic num_assistant_tokens_schedule

63c5f793

ArthurZucker requested a review from

gante 2 years ago

gante approved these changes on 2024-02-07

gante commented on 2024-02-07

gante requested a review from

amyeroberts 2 years ago

Update src/transformers/generation/configuration_utils.py

13275fc7

Update src/transformers/generation/candidate_generator.py

d0f84cc2

Update utils.py

58b98190

ArthurZucker commented on 2024-02-08

Update src/transformers/generation/candidate_generator.py

acdb235d

amyeroberts commented on 2024-02-07

Update tests/generation/test_utils.py

f7f2f4d9

make fixup

a248f1c4

merge conflict

ec79ba68

Merge branch 'fix_num_assistant_tokens' of https://github.com/jmamou/…

bfaccc06

fix docstring

52cde6bd

make fixup

04442ca2