I’m encountering an issue with the following line in fleur.py:
outputs = tokenizer.decode(output_ids[0, input_ids.shape[1]:]).strip()
In this case, input_ids has the shape torch.Size([1, 400]) and output_ids has the shape torch.Size([1, 7]). The slicing output_ids[0, input_ids.shape[1]:] results in an out-of-bounds operation, which causes outputs to be empty.