freewym's picture
fixes the asserion error when num_beams > 1
2258faa verified