x-transformers 2.4.0__py3-none-any.whl → 2.4.1__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 (5) hide show
  1. x_transformers/x_transformers.py +1 -1
  2. {x_transformers-2.4.0.dist-info → x_transformers-2.4.1.dist-info}/METADATA +1 -1
  3. {x_transformers-2.4.0.dist-info → x_transformers-2.4.1.dist-info}/RECORD +5 -5
  4. {x_transformers-2.4.0.dist-info → x_transformers-2.4.1.dist-info}/WHEEL +0 -0
  5. {x_transformers-2.4.0.dist-info → x_transformers-2.4.1.dist-info}/licenses/LICENSE +0 -0
x_transformers/x_transformers.py CHANGED
@@ -3263,7 +3263,7 @@ class TransformerWrapper(Module):
3263
3263
 
3264
3264
  # attention pool
3265
3265
 
3266
- if exists(self.attn_pool):
3266
+ if exists(self.attn_pool) and return_intermediates:
3267
3267
  queries = repeat(self.attn_pool_queries, 'n d -> b n d', b = x.shape[0])
3268
3268
 
3269
3269
  attn_pooled_tokens = self.attn_pool(queries, context = x, context_mask = mask)
{x_transformers-2.4.0.dist-info → x_transformers-2.4.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: x-transformers
3
- Version: 2.4.0
3
+ Version: 2.4.1
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.4.0.dist-info → x_transformers-2.4.1.dist-info}/RECORD RENAMED
@@ -8,10 +8,10 @@ x_transformers/entropy_based_tokenizer.py,sha256=F2lO8-v3aLIcVDVNhu7RR-UtRdlmaaY
8
8
  x_transformers/multi_input.py,sha256=tCh-fTJDj2ib4SMGtsa-AM8MxKzJAQSwqAXOu3HU2mg,9252
9
9
  x_transformers/neo_mlp.py,sha256=XCNnnop9WLarcxap1kGuYc1x8GHvwkZiDRnXOxSl3Po,3452
10
10
  x_transformers/nonautoregressive_wrapper.py,sha256=2NU58hYMgn-4Jzg3mie-mXb0XH_dCN7fjlzd3K1rLUY,10510
11
- x_transformers/x_transformers.py,sha256=IelVhLUuDmRnv6zXlQNvwUluW2RqVQQE2vYKCqctJyY,117583
11
+ x_transformers/x_transformers.py,sha256=9Fi0HvzpeIJqM6HlAd2M6JqsfjhTN1zEH9iFIimyjS4,117608
12
12
  x_transformers/xl_autoregressive_wrapper.py,sha256=CvZMJ6A6PA-Y_bQAhnORwjJBSl6Vjq2IdW5KTdk8NI8,4195
13
13
  x_transformers/xval.py,sha256=AwwYUm8yDAtKQyKJDIhYMsiLTJ_skh3scUFMjp5sda8,8597
14
- x_transformers-2.4.0.dist-info/METADATA,sha256=RyKkjmTnjbGUHA4EL-znJCPR17VF6i7ebvvgMKpTXVY,89896
15
- x_transformers-2.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
- x_transformers-2.4.0.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
17
- x_transformers-2.4.0.dist-info/RECORD,,
14
+ x_transformers-2.4.1.dist-info/METADATA,sha256=riX6ywwN305W2El7pkwY3GSCXrn5CNHFROTlcXN3yvo,89896
15
+ x_transformers-2.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
+ x_transformers-2.4.1.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
17
+ x_transformers-2.4.1.dist-info/RECORD,,