From f56f3d5468c90048ff324b801371ba03d82983bc Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 26 May 2022 08:44:20 -0600 Subject: [PATCH] Fix import issue for CVVP --- tortoise/models/cvvp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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):