summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/networks/conformer/subsampler.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/text_recognizer/networks/conformer/subsampler.py b/text_recognizer/networks/conformer/subsampler.py
index 42a983e..133b53a 100644
--- a/text_recognizer/networks/conformer/subsampler.py
+++ b/text_recognizer/networks/conformer/subsampler.py
@@ -4,9 +4,7 @@ from typing import Tuple
from einops import rearrange
from torch import nn, Tensor
-from text_recognizer.networks.transformer import (
- AxialPositionalEmbedding,
-)
+from text_recognizer.networks.transformer import AxialPositionalEmbedding
class Subsampler(nn.Module):