Add batched_next_token()
and batched_sample()
#1324
Loading
batched_next_token()
and batched_sample()
#1324