transformers
Speed up TopKLogitsWarper and TopPLogitsWarper (pytorch)
#9557

Merged

Speed up TopKLogitsWarper and TopPLogitsWarper (pytorch) #9557

sgugger merged 2 commits into huggingface:master from feature/speedup_top_filtering

make TopKLogitsWarper faster

f4c36c91

make TopPLogitsWarper faster

c0e30ca4

patrickvonplaten requested a review from

LysandreJik 5 years ago

patrickvonplaten requested a review from

sgugger 5 years ago

patrickvonplaten approved these changes on 2021-01-13

LysandreJik approved these changes on 2021-01-13

sgugger approved these changes on 2021-01-13

sgugger merged 0c9f01a8 into master 5 years ago

LSinev deleted the feature/speedup_top_filtering branch 4 years ago

Reviewers

LysandreJik

sgugger

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone