x-transformers 1.31.2__py3-none-any.whl → 1.31.3__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.
@@ -1403,7 +1403,7 @@ class AttentionLayers(Module):
1403
1403
 
1404
1404
  self.post_branch_fn_needs_condition = post_branch_fn_needs_condition
1405
1405
 
1406
- if not post_branch_fn_needs_condition and norm_add_unit_offset:
1406
+ if exists(post_branch_fn) and not post_branch_fn_needs_condition and norm_add_unit_offset:
1407
1407
  post_branch_fn = partial(post_branch_fn, unit_offset = 1.)
1408
1408
 
1409
1409
  # setup mlp for conditioning
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: x-transformers
3
- Version: 1.31.2
3
+ Version: 1.31.3
4
4
  Summary: X-Transformers - Pytorch
5
5
  Home-page: https://github.com/lucidrains/x-transformers
6
6
  Author: Phil Wang
@@ -4,11 +4,11 @@ x_transformers/autoregressive_wrapper.py,sha256=uX8Mb0zLsQrZECt_9UGt35g7tC05Rk3n
4
4
  x_transformers/continuous.py,sha256=WO52n9lFAXv5-SGadi2cApGF8dkouN8QSTEOuC7erj8,6180
5
5
  x_transformers/dpo.py,sha256=LjvWgCkqTl-UuehrzQ8nkX5guLr4whYwsmm7SKSwdls,3450
6
6
  x_transformers/nonautoregressive_wrapper.py,sha256=ys_p8obc7lTeeodCqvkRKxOXQ1C9T3j5Jwr-JbVgnXk,10432
7
- x_transformers/x_transformers.py,sha256=aeFAk19BRxkLOS-pUgtDk3HaMS6_dsuJtz2o_KkUkOU,75805
7
+ x_transformers/x_transformers.py,sha256=D9l3jL1D0RzHynIJcSUpxQec1n-7cHgRQZNDdDoYCFQ,75832
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.31.2.dist-info/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
11
- x_transformers-1.31.2.dist-info/METADATA,sha256=dxDZYzaZwwOCD-9I6ver-SSLlSG4diss-ovDo2H_9xE,661
12
- x_transformers-1.31.2.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
13
- x_transformers-1.31.2.dist-info/top_level.txt,sha256=hO6KGpFuGucRNEtRfme4A_rGcM53AKwGP7RVlRIxS5Q,15
14
- x_transformers-1.31.2.dist-info/RECORD,,
10
+ x_transformers-1.31.3.dist-info/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
11
+ x_transformers-1.31.3.dist-info/METADATA,sha256=cS3vVeEi3fSoYqXWDkW6FJhgmX11vrfQr9Lc3kMT9MI,661
12
+ x_transformers-1.31.3.dist-info/WHEEL,sha256=mguMlWGMX-VHnMpKOjjQidIo1ssRlCFu4a4mBpz1s2M,91
13
+ x_transformers-1.31.3.dist-info/top_level.txt,sha256=hO6KGpFuGucRNEtRfme4A_rGcM53AKwGP7RVlRIxS5Q,15
14
+ x_transformers-1.31.3.dist-info/RECORD,,