tests/generation/test_logits_process.py · 4df1d696340794cc9c03f4d14482375207f1e7a7 · chenpangpang / transformers

[BUG] BarkEosPrioritizerLogitsProcessor eos_token_id use list, tensor size mismatch (#28201) · 4df1d696

HanHui authored Jan 10, 2024



fix(generation/logits_process.py): BarkEosPrioritizerLogitsProcessor eos_token_id use list, tensor size mismatch
Co-authored-by: chenhanhui <chenhanhui@kanzhun.com>

4df1d696

test_logits_process.py 36.1 KB

Replace test_logits_process.py