remove xt dep

This commit is contained in:
James Betker 2022-04-20 18:00:12 -06:00
parent ca31a1c9f7
commit 2347d16333

View File

@ -13,8 +13,6 @@ from einops.layers.torch import Rearrange
from entmax import entmax15 from entmax import entmax15
from torch.utils.checkpoint import checkpoint from torch.utils.checkpoint import checkpoint
from x_transformers.autoregressive_wrapper import AutoregressiveWrapper
DEFAULT_DIM_HEAD = 64 DEFAULT_DIM_HEAD = 64
Intermediates = namedtuple('Intermediates', [ Intermediates = namedtuple('Intermediates', [