diff --git a/tortoise/models/cvvp.py b/tortoise/models/cvvp.py
index 0cce0ae..622dd60 100644
--- a/tortoise/models/cvvp.py
+++ b/tortoise/models/cvvp.py
@@ -4,8 +4,8 @@ import torch.nn.functional as F
 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):