Fix import issue for CVVP

del18687058912 · May 26, 2022 · f56f3d5 · f56f3d5
1 parent 3acca14
commit f56f3d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tortoise/models/cvvp.py b/tortoise/models/cvvp.py
@@ -4,8 +4,8 @@
 from torch import einsum
 from torch.utils.checkpoint import checkpoint
 
-from models.arch_util import AttentionBlock
-from models.xtransformers import ContinuousTransformerWrapper, Encoder
+from tortoise.models.arch_util import AttentionBlock
+from tortoise.models.xtransformers import ContinuousTransformerWrapper, Encoder
 
 
 def exists(val):