x-transformers 2.3.12__py3-none-any.whl → 2.3.14__py3-none-any.whl

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 (6) hide show
  1. x_transformers/attend.py +1 -1
  2. x_transformers/xval.py +1 -1
  3. {x_transformers-2.3.12.dist-info → x_transformers-2.3.14.dist-info}/METADATA +1 -1
  4. {x_transformers-2.3.12.dist-info → x_transformers-2.3.14.dist-info}/RECORD +6 -6
  5. {x_transformers-2.3.12.dist-info → x_transformers-2.3.14.dist-info}/WHEEL +0 -0
  6. {x_transformers-2.3.12.dist-info → x_transformers-2.3.14.dist-info}/licenses/LICENSE +0 -0
x_transformers/attend.py CHANGED
@@ -165,7 +165,7 @@ class Attend(Module):
165
165
  post_talking_heads = False,
166
166
  pre_scale_post_talking_heads = False,
167
167
  sparse_topk = None,
168
- sparse_topk_straight_through = False,
168
+ sparse_topk_straight_through = False, # https://arxiv.org/abs/2505.22074
169
169
  scale = None,
170
170
  qk_norm = False,
171
171
  l2_distance = False,
x_transformers/xval.py CHANGED
@@ -11,7 +11,7 @@ import torch.nn.functional as F
11
11
  from typing import Callable
12
12
  from collections import namedtuple
13
13
 
14
- from einops import rearrange
14
+ from einops import rearrange, repeat, pack, unpack
15
15
  from einops.layers.torch import Rearrange
16
16
 
17
17
  from x_transformers.x_transformers import (
{x_transformers-2.3.12.dist-info → x_transformers-2.3.14.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: x-transformers
3
- Version: 2.3.12
3
+ Version: 2.3.14
4
4
  Summary: X-Transformers
5
5
  Project-URL: Homepage, https://pypi.org/project/x-transformers/
6
6
  Project-URL: Repository, https://github.com/lucidrains/x-transformers
{x_transformers-2.3.12.dist-info → x_transformers-2.3.14.dist-info}/RECORD RENAMED
@@ -1,5 +1,5 @@
1
1
  x_transformers/__init__.py,sha256=h3I2ejobgEdy8H7NgV-rP8UaBCnd16-MysvDXH9GMEA,985
2
- x_transformers/attend.py,sha256=-5BWWhFsp7tvZTdN91Ay5SqOjyj9uOs-122vFvoO6b4,17253
2
+ x_transformers/attend.py,sha256=xFsBtl7h7_qebPh7kE81BpmCWAjCgFpB9i_IHu_91es,17288
3
3
  x_transformers/autoregressive_wrapper.py,sha256=reLCno9Z9pchVU79tBF8OMo21LwSZ67KAeB83jqkyAc,10505
4
4
  x_transformers/belief_state_wrapper.py,sha256=YLUMk6t2MhFBEw5lHDDHJHcoCxTIkHvxTNY__GGZEKU,13374
5
5
  x_transformers/continuous.py,sha256=KPKi7TKqHYcDWYVhSkSB9y5iZMnhzVZxHhjJRdL7w5I,9521
@@ -10,8 +10,8 @@ x_transformers/neo_mlp.py,sha256=XCNnnop9WLarcxap1kGuYc1x8GHvwkZiDRnXOxSl3Po,345
10
10
  x_transformers/nonautoregressive_wrapper.py,sha256=2NU58hYMgn-4Jzg3mie-mXb0XH_dCN7fjlzd3K1rLUY,10510
11
11
  x_transformers/x_transformers.py,sha256=ZfOXrZSiy2jlZ8wVmDdMTLW4hAY_qfmPQHW9t2ABxbo,114097
12
12
  x_transformers/xl_autoregressive_wrapper.py,sha256=CvZMJ6A6PA-Y_bQAhnORwjJBSl6Vjq2IdW5KTdk8NI8,4195
13
- x_transformers/xval.py,sha256=7S00kCuab4tWQa-vf-z-XfzADjVj48MoFIr7VSIvttg,8575
14
- x_transformers-2.3.12.dist-info/METADATA,sha256=ADkMqVrlqcYcMk8U_9oxqOHc4XRD1NbSfVJsGhy5R8s,89022
15
- x_transformers-2.3.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
- x_transformers-2.3.12.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
17
- x_transformers-2.3.12.dist-info/RECORD,,
13
+ x_transformers/xval.py,sha256=AwwYUm8yDAtKQyKJDIhYMsiLTJ_skh3scUFMjp5sda8,8597
14
+ x_transformers-2.3.14.dist-info/METADATA,sha256=Tnvnrfnr-eIlUVEH3IePLykynVikAq-t01v4pSh3yPQ,89022
15
+ x_transformers-2.3.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
+ x_transformers-2.3.14.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
17
+ x_transformers-2.3.14.dist-info/RECORD,,