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 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', [
|
||||||
|
|
Loading…
Reference in New Issue
Block a user