quick fix

lucidrains · Jul 17, 2023 · 3451614 · 3451614
1 parent c6b6d2e
commit 3451614
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'x-transformers',
   packages = find_packages(exclude=['examples']),
-  version = '1.16.20',
+  version = '1.16.21',
   license='MIT',
   description = 'X-Transformers - Pytorch',
   author = 'Phil Wang',

diff --git a/x_transformers/x_transformers.py b/x_transformers/x_transformers.py
@@ -1456,7 +1456,7 @@ def forward(
         x,
         return_embeddings = False,
         return_intermediates = False,
-        return_mems = True,
+        return_mems = False,
         mask = None,
         return_attn = False,
         mems = None,