generator.py, line 31, in step candidate = prev_candidates[b_ix] + [t_ix]. TypeError: list indices must be integers or slices, not float

[AegeanYan]

Hello, I'm interested in your work. But when I'm running the run.sh, I found this error and it seems that the beam_ix and token_ix in generator.py is float, not int. For example, the beam_ix is tensor([0.8333, 0.1667, 0.5000, 0.0000, 0.3333, 0.6667], device='cuda:0'), I wonder how I can fix it. Thanks