Updated imports

This commit is contained in:
Christian Risi 2025-10-06 15:45:28 +02:00
parent a693cbb77e
commit 49f0beb6ea
3 changed files with 3 additions and 1 deletions

View File

@ -3,11 +3,13 @@ from .Encoder import Encoder
from .FeedForwardNetwork import FeedForwardNetwork from .FeedForwardNetwork import FeedForwardNetwork
from .MultiHeadAttention import MultiheadAttention from .MultiHeadAttention import MultiheadAttention
from .TorchMultiHeadAttention import TorchMultiHeadAttention from .TorchMultiHeadAttention import TorchMultiHeadAttention
from .SpannedMasker import SpannedMasker
__all__ = [ __all__ = [
"Decoder", "Decoder",
"Encoder", "Encoder",
"FeedForwardNetwork", "FeedForwardNetwork",
"MultiheadAttention", "MultiheadAttention",
"TorchMultiHeadAttention" "TorchMultiHeadAttention",
"SpannedMasker"
] ]