Commit Graph

3 Commits

Author SHA1 Message Date
Byron Hsu
17e998f1a8 Test consistency for single and batch seperately (#1590) 2024-10-06 22:02:27 +00:00
Ying Sheng
c98e84c21e [Minor, Performance] Use torch.argmax for greedy sampling (#1589) 2024-10-06 13:15:05 -07:00
Ke Bao
33b54e7c40 Add pytorch sampling backend ut (#1425) 2024-09-15 01:15:30 +10:00