mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make performer code functional

Browse Source
This commit is contained in:
James Betker 2022-01-09 22:32:50 -07:00
parent f474a7ac65
commit 432073c5ca
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/lucidrains/performer/__init__.py
View File

@ -1,3 +1 @@
from performer_pytorch import PerformerLM, Performer, FastAttention, SelfAttention, CrossAttention, ProjectionUpdater
from autoregressive_wrapper import AutoregressiveWrapper
from performer_enc_dec import PerformerEncDec

2
codes/models/lucidrains/performer/performer_pytorch.py
View File

@ -10,7 +10,7 @@ from contextlib import contextmanager
from local_attention import LocalAttention from local_attention import LocalAttention
from axial_positional_embedding import AxialPositionalEmbedding from axial_positional_embedding import AxialPositionalEmbedding
from reversible import ReversibleSequence, SequentialSequence from models.lucidrains.performer.reversible import ReversibleSequence, SequentialSequence
from distutils.version import LooseVersion from distutils.version import LooseVersion