From 432073c5ca8beedc5fd0e893cc835c2a72629b05 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 9 Jan 2022 22:32:50 -0700 Subject: [PATCH] Make performer code functional --- codes/models/lucidrains/performer/__init__.py | 4 +--- codes/models/lucidrains/performer/performer_pytorch.py | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/codes/models/lucidrains/performer/__init__.py b/codes/models/lucidrains/performer/__init__.py index 0dfe9687..8b137891 100644 --- a/codes/models/lucidrains/performer/__init__.py +++ b/codes/models/lucidrains/performer/__init__.py @@ -1,3 +1 @@ -from performer_pytorch import PerformerLM, Performer, FastAttention, SelfAttention, CrossAttention, ProjectionUpdater -from autoregressive_wrapper import AutoregressiveWrapper -from performer_enc_dec import PerformerEncDec + diff --git a/codes/models/lucidrains/performer/performer_pytorch.py b/codes/models/lucidrains/performer/performer_pytorch.py index 836ed03f..2d618f11 100644 --- a/codes/models/lucidrains/performer/performer_pytorch.py +++ b/codes/models/lucidrains/performer/performer_pytorch.py @@ -10,7 +10,7 @@ from contextlib import contextmanager from local_attention import LocalAttention from axial_positional_embedding import AxialPositionalEmbedding -from reversible import ReversibleSequence, SequentialSequence +from models.lucidrains.performer.reversible import ReversibleSequence, SequentialSequence from distutils.version import LooseVersion