From c7e5354ffa43eccfc4e411375ce2f531af7bbcff Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 15 Apr 2024 21:49:02 +0200 Subject: Format --- text_recognizer/network/mammut.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/text_recognizer/network/mammut.py b/text_recognizer/network/mammut.py index 761e5f9..8754420 100644 --- a/text_recognizer/network/mammut.py +++ b/text_recognizer/network/mammut.py @@ -1,12 +1,12 @@ from typing import Type -from einops import repeat import torch -from torch import nn, Tensor import torch.nn.functional as F +from einops import repeat +from torch import Tensor, nn -from text_recognizer.network.transformer.decoder import Decoder from text_recognizer.network.transformer.attention import Attention +from text_recognizer.network.transformer.decoder import Decoder class ToLatent(nn.Module): -- cgit v1.2.3-70-g09d2