x-transformers 1.37.1__py3-none-any.whl → 1.37.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.
- x_transformers/x_transformers.py +1 -1
- {x_transformers-1.37.1.dist-info → x_transformers-1.37.2.dist-info}/METADATA +1 -1
- {x_transformers-1.37.1.dist-info → x_transformers-1.37.2.dist-info}/RECORD +6 -6
- {x_transformers-1.37.1.dist-info → x_transformers-1.37.2.dist-info}/LICENSE +0 -0
- {x_transformers-1.37.1.dist-info → x_transformers-1.37.2.dist-info}/WHEEL +0 -0
- {x_transformers-1.37.1.dist-info → x_transformers-1.37.2.dist-info}/top_level.txt +0 -0
x_transformers/x_transformers.py
CHANGED
@@ -2258,7 +2258,7 @@ class TransformerWrapper(Module):
|
|
2258
2258
|
# attention layers
|
2259
2259
|
|
2260
2260
|
if not self.recycling:
|
2261
|
-
assert recycle_steps == 1, 'you did not train with recycling'
|
2261
|
+
assert not exists(recycle_steps) or recycle_steps == 1, 'you did not train with recycling'
|
2262
2262
|
|
2263
2263
|
# regular
|
2264
2264
|
|
@@ -5,11 +5,11 @@ x_transformers/continuous.py,sha256=cIVEdhfei258__ziV7kQBrJMxCel54bExBTDrO9rfCI,
|
|
5
5
|
x_transformers/dpo.py,sha256=LjvWgCkqTl-UuehrzQ8nkX5guLr4whYwsmm7SKSwdls,3450
|
6
6
|
x_transformers/multi_input.py,sha256=tCh-fTJDj2ib4SMGtsa-AM8MxKzJAQSwqAXOu3HU2mg,9252
|
7
7
|
x_transformers/nonautoregressive_wrapper.py,sha256=ys_p8obc7lTeeodCqvkRKxOXQ1C9T3j5Jwr-JbVgnXk,10432
|
8
|
-
x_transformers/x_transformers.py,sha256=
|
8
|
+
x_transformers/x_transformers.py,sha256=1VoPI4RhMWjRMI9THucMc7ElM-8E-8rA31oc9KlpWPw,83385
|
9
9
|
x_transformers/xl_autoregressive_wrapper.py,sha256=DCx4n0_c1tFai4nOqaWVnqx2p9eutsZsDMiMP1ckxNU,4117
|
10
10
|
x_transformers/xval.py,sha256=QE1ltYZTR_eGgIHPP2BrMWVWVLqMW-OpDZh87BSmQEg,8563
|
11
|
-
x_transformers-1.37.
|
12
|
-
x_transformers-1.37.
|
13
|
-
x_transformers-1.37.
|
14
|
-
x_transformers-1.37.
|
15
|
-
x_transformers-1.37.
|
11
|
+
x_transformers-1.37.2.dist-info/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
|
12
|
+
x_transformers-1.37.2.dist-info/METADATA,sha256=6VSLhVR_d0rpsBGny6FteK8XVddFmQ77W1TzeGcVWxs,661
|
13
|
+
x_transformers-1.37.2.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
|
14
|
+
x_transformers-1.37.2.dist-info/top_level.txt,sha256=hO6KGpFuGucRNEtRfme4A_rGcM53AKwGP7RVlRIxS5Q,15
|
15
|
+
x_transformers-1.37.2.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|