4 Commits

Author	SHA1	Message	Date
Lianmin Zheng	ad4125d1a9	Fuse more ops & Simplify token mapping (#1758)	2024-10-22 23:20:43 -07:00
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