remove dependency on x-transformers

This commit is contained in:
James Betker 2022-04-18 20:43:04 -06:00
parent 078274f055
commit 412a441c86
2 changed files with 2 additions and 4 deletions

View File

@ -5,8 +5,7 @@ import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
import torchaudio import torchaudio
from x_transformers import ContinuousTransformerWrapper from models.xtransformers import ContinuousTransformerWrapper, RelativePositionBias
from x_transformers.x_transformers import RelativePositionBias
def zero_module(module): def zero_module(module):

View File

@ -6,5 +6,4 @@ tokenizers
inflect inflect
progressbar progressbar
einops einops
unidecode unidecode
x-transformers