summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/models/vqgan.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/models/vqgan.py b/text_recognizer/models/vqgan.py
index 6e4d3fb..6a90e06 100644
--- a/text_recognizer/models/vqgan.py
+++ b/text_recognizer/models/vqgan.py
@@ -4,8 +4,8 @@ from typing import Tuple
import attr
from torch import Tensor
+from text_recognizer.criterion.vqgan_loss import VQGANLoss
from text_recognizer.models.base import BaseLitModel
-from text_recognizer.criterions.vqgan_loss import VQGANLoss
@attr.s(auto_attribs=True, eq=False)