-
Notifications
You must be signed in to change notification settings - Fork 3
Open
Description
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.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels