From cfb460666953c87f606833bf597b53eba0a2900d Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 5 Sep 2022 00:06:37 +0200 Subject: Format --- text_recognizer/models/__init__.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'text_recognizer/models/__init__.py') diff --git a/text_recognizer/models/__init__.py b/text_recognizer/models/__init__.py index cc02487..56e3e93 100644 --- a/text_recognizer/models/__init__.py +++ b/text_recognizer/models/__init__.py @@ -1,2 +1,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 -- cgit v1.2.3-70-g09d2