remove pytorch-fast-transformers from dep

lucidrains · Apr 21, 2021 · 457dade · 457dade
1 parent e40abb6
commit 457dade
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -12,6 +12,12 @@ An implementation of <a href="https://arxiv.org/abs/2009.14794">Performer</a>, a
 $ pip install performer-pytorch
 ```
 
+Then you must run the following, if you plan on training an autoregressive model
+
+```bash
+$ pip install -r requirements.txt
+```
+
 ## Usage
 
 Performer Language Model

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1 @@
+pytorch-fast-transformers>=0.3.0
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'performer-pytorch',
   packages = find_packages(exclude=['examples']),
-  version = '1.0.10',
+  version = '1.0.11',
   license='MIT',
   description = 'Performer - Pytorch',
   author = 'Phil Wang',
@@ -18,7 +18,6 @@
   install_requires=[
     'einops>=0.3',
     'local-attention>=1.1.1',
-    'pytorch-fast-transformers>=0.3.0',
     'torch>=1.6',
     'axial-positional-embedding>=0.1.0'
   ],