remove xt dep
This commit is contained in:
parent
0b496a0a38
commit
053b8d138a
|
@ -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', [
|
||||
|
|
Loading…
Reference in New Issue
Block a user