blob: 73c18d15bbf6d8710dd490d18b6c7004025f43d6 (
plain)
1
2
3
4
5
6
7
8
9
|
"""Transformer modules."""
from text_recognizer.networks.transformer.attention import Attention
from text_recognizer.networks.transformer.ff import FeedForward
from text_recognizer.networks.transformer.norm import RMSNorm
from text_recognizer.networks.transformer.decoder import Decoder, DecoderBlock
from text_recognizer.networks.transformer.embeddings.rotary import RotaryEmbedding
from text_recognizer.networks.transformer.embeddings.axial import (
AxialPositionalEmbedding,
)
|
