diff options
Diffstat (limited to 'text_recognizer/models')
| -rw-r--r-- | text_recognizer/models/vqgan.py | 2 |
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) |