From 053b8d138a5368c0bfadfd61ba6f77ce22589464 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 20 Apr 2022 18:00:12 -0600 Subject: [PATCH] remove xt dep --- models/xtransformers.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/models/xtransformers.py b/models/xtransformers.py index 19bf795..70e8e63 100644 --- a/models/xtransformers.py +++ b/models/xtransformers.py @@ -13,8 +13,6 @@ from einops.layers.torch import Rearrange from entmax import entmax15 from torch.utils.checkpoint import checkpoint -from x_transformers.autoregressive_wrapper import AutoregressiveWrapper - DEFAULT_DIM_HEAD = 64 Intermediates = namedtuple('Intermediates', [