From 1a686963adb57118d5641adebf5185bfacaea015 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 27 Oct 2021 22:42:10 +0200
Subject: Fix import of criterion in VQGAN

---
 text_recognizer/models/vqgan.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'text_recognizer/models/vqgan.py')

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)
-- 
cgit v1.2.3-70-g09d2