Update pre-commit & run on all files

This commit is contained in:
Simon Alibert
2024-03-25 17:29:35 +01:00
parent 0fae5b206b
commit bcfdba109f
6 changed files with 7 additions and 3 deletions

View File

@@ -6,6 +6,7 @@ Copy-paste from torch.nn.Transformer with modifications:
* extra LN at the end of encoder is removed
* decoder returns a stack of activations from all decoding layers
"""
import copy
from typing import Optional