transformers
0baa9246 - Fix TypicalLogitsWarper tensor OOB indexing edge case (#26579)

Commit
2 years ago
Fix TypicalLogitsWarper tensor OOB indexing edge case (#26579) * Fix TypicalLogitsWarper tensor OOB indexing edge case This can be triggerd fairly quickly with low precision e.g. bfloat16 and typical_p = 0.99. * Shift threshold index by one * Use explicit named arg for clamp min
Author
Parents
Loading