Fix import issue for CVVP

This commit is contained in:
James Betker 2022-05-26 08:44:20 -06:00
parent 3acca1445a
commit f56f3d5468

View File

@ -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):