transformers
[BUG] BarkEosPrioritizerLogitsProcessor eos_token_id use list, tensor size mismatch
#28201

Merged

[BUG] BarkEosPrioritizerLogitsProcessor eos_token_id use list, tensor size mismatch #28201

ArthurZucker merged 1 commit into huggingface:main from inkinworld:fix-barkeos-processor

inkinworld force pushed 2 years ago

fix(generation/logits_process.py): BarkEosPrioritizerLogitsProcessor …

ce52befe

inkinworld force pushed to ce52befe 2 years ago

gante approved these changes on 2024-01-09

gante requested a review from

ArthurZucker 2 years ago

ArthurZucker approved these changes on 2024-01-10

ArthurZucker merged 4df1d696 into main 2 years ago

Reviewers

ArthurZucker

gante

Assignees

No one assigned

Labels

None yet

Milestone

No milestone