Skip to content

Commit 88bfa98

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent e10941a commit 88bfa98

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

computer_vision/vision_transformer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,9 @@ def feedforward_network(x: np.ndarray, hidden_dim: int = 3072) -> np.ndarray:
285285

286286

287287
def transformer_encoder_block(
288-
x: np.ndarray, num_heads: int = 12, hidden_dim: int = 3072 # noqa: ARG001
288+
x: np.ndarray,
289+
num_heads: int = 12,
290+
hidden_dim: int = 3072, # noqa: ARG001
289291
) -> np.ndarray:
290292
"""
291293
Apply a single Transformer encoder block.

0 commit comments

Comments
 (0)