x-transformers 1.32.1__py3-none-any.whl → 1.32.2__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 +2 -2
  2. {x_transformers-1.32.1.dist-info → x_transformers-1.32.2.dist-info}/METADATA +1 -1
  3. {x_transformers-1.32.1.dist-info → x_transformers-1.32.2.dist-info}/RECORD +6 -6
  4. {x_transformers-1.32.1.dist-info → x_transformers-1.32.2.dist-info}/LICENSE +0 -0
  5. {x_transformers-1.32.1.dist-info → x_transformers-1.32.2.dist-info}/WHEEL +0 -0
  6. {x_transformers-1.32.1.dist-info → x_transformers-1.32.2.dist-info}/top_level.txt +0 -0
x_transformers/attend.py CHANGED
@@ -241,7 +241,7 @@ class Attend(Module):
241
241
 
242
242
  # for a row that is entirely masked out, should zero out the output of that row token
243
243
 
244
- if exists(row_is_entirely_masked):
244
+ if exists(row_is_entirely_masked) and row_is_entirely_masked.any():
245
245
  out = out.masked_fill(row_is_entirely_masked[..., None], 0.)
246
246
 
247
247
  return out, Intermediates()
@@ -361,7 +361,7 @@ class Attend(Module):
361
361
  post_softmax_attn = post_softmax_attn
362
362
  )
363
363
 
364
- if exists(row_is_entirely_masked):
364
+ if exists(row_is_entirely_masked) and row_is_entirely_masked.any():
365
365
  out = out.masked_fill(row_is_entirely_masked[..., None], 0.)
366
366
 
367
367
  return out, intermediates
{x_transformers-1.32.1.dist-info → x_transformers-1.32.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: x-transformers
3
- Version: 1.32.1
3
+ Version: 1.32.2
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.1.dist-info → x_transformers-1.32.2.dist-info}/RECORD RENAMED
@@ -1,5 +1,5 @@
1
1
  x_transformers/__init__.py,sha256=5ms39Df8osTUHQ-XTCgP4vSUA4UiNpim9VXJtrLrIvQ,724
2
- x_transformers/attend.py,sha256=PzJ0_MnysNDV-jcDuCbAFPEUZxgljVYOShvtMGfO7wU,11283
2
+ x_transformers/attend.py,sha256=MI-m91wumBFqFqr_KK9MLgsLk_vPeaVbFMyDr_mWdmY,11349
3
3
  x_transformers/autoregressive_wrapper.py,sha256=uX8Mb0zLsQrZECt_9UGt35g7tC05Rk3nPqO6xp2FFCc,9619
4
4
  x_transformers/continuous.py,sha256=WO52n9lFAXv5-SGadi2cApGF8dkouN8QSTEOuC7erj8,6180
5
5
  x_transformers/dpo.py,sha256=LjvWgCkqTl-UuehrzQ8nkX5guLr4whYwsmm7SKSwdls,3450
@@ -7,8 +7,8 @@ x_transformers/nonautoregressive_wrapper.py,sha256=ys_p8obc7lTeeodCqvkRKxOXQ1C9T
7
7
  x_transformers/x_transformers.py,sha256=1QG7zUe89h1R5VDMoKEAkvdRRDkzQ7h6npkqblxxR6g,76312
8
8
  x_transformers/xl_autoregressive_wrapper.py,sha256=DCx4n0_c1tFai4nOqaWVnqx2p9eutsZsDMiMP1ckxNU,4117
9
9
  x_transformers/xval.py,sha256=QE1ltYZTR_eGgIHPP2BrMWVWVLqMW-OpDZh87BSmQEg,8563
10
- x_transformers-1.32.1.dist-info/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
11
- x_transformers-1.32.1.dist-info/METADATA,sha256=NAyDRgkGJJ5ovqj1--2CgHKalfczAxWcyKiEr7Spxc4,661
12
- x_transformers-1.32.1.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
13
- x_transformers-1.32.1.dist-info/top_level.txt,sha256=hO6KGpFuGucRNEtRfme4A_rGcM53AKwGP7RVlRIxS5Q,15
14
- x_transformers-1.32.1.dist-info/RECORD,,
10
+ x_transformers-1.32.2.dist-info/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
11
+ x_transformers-1.32.2.dist-info/METADATA,sha256=U0Kh4e7UiL-0hLDZb0P3McdvTnzTeFyVwtoXFffzQ-M,661
12
+ x_transformers-1.32.2.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
13
+ x_transformers-1.32.2.dist-info/top_level.txt,sha256=hO6KGpFuGucRNEtRfme4A_rGcM53AKwGP7RVlRIxS5Q,15
14
+ x_transformers-1.32.2.dist-info/RECORD,,