x-transformers 1.32.11__tar.gz → 1.32.12__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (21) hide show
  1. {x_transformers-1.32.11/x_transformers.egg-info → x_transformers-1.32.12}/PKG-INFO +1 -1
  2. {x_transformers-1.32.11 → x_transformers-1.32.12}/setup.py +1 -1
  3. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/x_transformers.py +1 -1
  4. {x_transformers-1.32.11 → x_transformers-1.32.12/x_transformers.egg-info}/PKG-INFO +1 -1
  5. {x_transformers-1.32.11 → x_transformers-1.32.12}/LICENSE +0 -0
  6. {x_transformers-1.32.11 → x_transformers-1.32.12}/README.md +0 -0
  7. {x_transformers-1.32.11 → x_transformers-1.32.12}/setup.cfg +0 -0
  8. {x_transformers-1.32.11 → x_transformers-1.32.12}/tests/test_x_transformers.py +0 -0
  9. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/__init__.py +0 -0
  10. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/attend.py +0 -0
  11. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/autoregressive_wrapper.py +0 -0
  12. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/continuous.py +0 -0
  13. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/dpo.py +0 -0
  14. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/multi_input.py +0 -0
  15. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/nonautoregressive_wrapper.py +0 -0
  16. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/xl_autoregressive_wrapper.py +0 -0
  17. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/xval.py +0 -0
  18. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers.egg-info/SOURCES.txt +0 -0
  19. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers.egg-info/dependency_links.txt +0 -0
  20. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers.egg-info/requires.txt +0 -0
  21. {x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers.egg-info/top_level.txt +0 -0
{x_transformers-1.32.11/x_transformers.egg-info → x_transformers-1.32.12}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: x-transformers
3
- Version: 1.32.11
3
+ Version: 1.32.12
4
4
  Summary: X-Transformers - Pytorch
5
5
  Home-page: https://github.com/lucidrains/x-transformers
6
6
  Author: Phil Wang
{x_transformers-1.32.11 → x_transformers-1.32.12}/setup.py RENAMED
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
3
3
  setup(
4
4
  name = 'x-transformers',
5
5
  packages = find_packages(exclude=['examples']),
6
- version = '1.32.11',
6
+ version = '1.32.12',
7
7
  license='MIT',
8
8
  description = 'X-Transformers - Pytorch',
9
9
  author = 'Phil Wang',
{x_transformers-1.32.11 → x_transformers-1.32.12}/x_transformers/x_transformers.py RENAMED
@@ -2184,7 +2184,7 @@ class TransformerWrapper(Module):
2184
2184
  # maybe squeeze out last dimension of logits
2185
2185
 
2186
2186
  if self.squeeze_out_last_dim:
2187
- logits = tuple(rearrange(t, '... 1 -> ...') for t in cast_tuple(logits))
2187
+ logits = tuple((rearrange(t, '... 1 -> ...') if t.shape[-1] == 1 else t) for t in cast_tuple(logits))
2188
2188
 
2189
2189
  if not self.has_multiple_heads:
2190
2190
  logits = first(logits)
{x_transformers-1.32.11 → x_transformers-1.32.12/x_transformers.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: x-transformers
3
- Version: 1.32.11
3
+ Version: 1.32.12
4
4
  Summary: X-Transformers - Pytorch
5
5
  Home-page: https://github.com/lucidrains/x-transformers
6
6
  Author: Phil Wang