text_recognizer/networks/transformer/__init__.py


1
2
3
4
5
6

"""Transformer modules."""
from text_recognizer.networks.transformer.embeddings.rotary import RotaryEmbedding
from text_recognizer.networks.transformer.attention import Attention
from text_recognizer.networks.transformer.embeddings.axial import (
    AxialPositionalEmbedding,
)