DL-Art-School/codes/models/lucidrains/performer/__init__.py

4 lines
221 B
Python
Raw Normal View History

2022-01-10 05:10:07 +00:00
from performer_pytorch import PerformerLM, Performer, FastAttention, SelfAttention, CrossAttention, ProjectionUpdater
from autoregressive_wrapper import AutoregressiveWrapper
from performer_enc_dec import PerformerEncDec