summaryrefslogtreecommitdiff
path: root/text_recognizer/models/__init__.py
blob: 56e3e93daf8e04654814259988d33b4cdaada7d3 (plain)
1
2
3
4
"""PyTorch Lightning models modules."""
from text_recognizer.models.transformer import LitTransformer
from text_recognizer.models.perceiver import LitPerceiver
from text_recognizer.models.vq_transformer import LitVqTransformer