Fix Typo (#24559)

63cc30e7 · MS Kim(tony9402) · GitHub · ae454f41 · 63cc30e7
Unverified Commit 63cc30e7 authored Jun 29, 2023 by MS Kim(tony9402) Committed by GitHub Jun 29, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/fsmt/modeling_fsmt.py src/transformers/models/fsmt/modeling_fsmt.py +1 -1

No files found.
--- a/src/transformers/models/fsmt/modeling_fsmt.py
+++ b/src/transformers/models/fsmt/modeling_fsmt.py
@@ -770,7 +770,7 @@ class FSMTDecoder(nn.Module):
        x += positions
        x = nn.functional.dropout(x, p=self.dropout, training=self.training)

-        # Convert to FSMT output format: (seq_len, BS, model_dim) -> (BS, seq_len, model_dim)
+        # Convert to FSMT output format: (BS, seq_len, model_dim) -> (seq_len, BS, model_dim)
        x = x.transpose(0, 1)
        encoder_hidden_states = encoder_hidden_states.transpose(0, 1)