In the batch size 2 example you have 226 time steps in log probs and 78 <= 226.
In the batch size 1 example you have 42 time steps in log probs and 77 > 42.
You cannot feed fewer time steps than your input size claims the input has.
Best regards
Thomas