Fix Typo

[tony9402]

What does this PR do?

Fixed wrong annotation

• (seq_len, BS, model_dim) -> (BS, seq_len, model_dim) -> (BS, seq_len, model_dim) -> (seq_len, BS, model_dim)

Before submitting

• This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
• Did you read the contributor guideline,
  Pull Request section?

[sgugger]

Can you check if this is correct @ArthurZucker ?

[HuggingFaceDocBuilderDev]

The documentation is not available anymore as the PR was closed or merged.

[ArthurZucker]

LGTM as their is a similar comment afterwards, which puts it back to transformesr format:

# Convert to standard output format: (seq_len, BS, model_dim) -> (BS, seq_len, model_dim)