remove dependency on x-transformers
This commit is contained in:
parent
ad0f3fdd58
commit
24a5b840ae
|
@ -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):
|
||||||
|
|
|
@ -7,4 +7,3 @@ inflect
|
||||||
progressbar
|
progressbar
|
||||||
einops
|
einops
|
||||||
unidecode
|
unidecode
|
||||||
x-transformers
|
|
Loading…
Reference in New Issue
Block a user