Update attention.py

apple · Jan 24, 2025 · 20a1b4c · 20a1b4c
1 parent 42f52cd
commit 20a1b4c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/axlearn/common/attention.py b/axlearn/common/attention.py
@@ -1245,8 +1245,10 @@ def forward(
         cfg = self.config
         if positions is not None and max_seq_len is not None:
             if max_seq_len != positions.shape[-1]:
-                raise ValueError("Both `positions` and `max_seq_len` are provided and they "
-                                 "do not match. You only need to provide one of them.")
+                raise ValueError(
+                    "Both `positions` and `max_seq_len` are provided and they "
+                    "do not match. You only need to provide one of them."
+                )
         if positions is None:
             if max_seq_len is None:
                 raise ValueError(