x-transformers 1.28.0__py3-none-any.whl → 1.28.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.
- x_transformers/x_transformers.py +1 -1
- {x_transformers-1.28.0.dist-info → x_transformers-1.28.1.dist-info}/METADATA +1 -1
- {x_transformers-1.28.0.dist-info → x_transformers-1.28.1.dist-info}/RECORD +6 -6
- {x_transformers-1.28.0.dist-info → x_transformers-1.28.1.dist-info}/LICENSE +0 -0
- {x_transformers-1.28.0.dist-info → x_transformers-1.28.1.dist-info}/WHEEL +0 -0
- {x_transformers-1.28.0.dist-info → x_transformers-1.28.1.dist-info}/top_level.txt +0 -0
x_transformers/x_transformers.py
CHANGED
@@ -946,7 +946,7 @@ class Attention(nn.Module):
|
|
946
946
|
|
947
947
|
# append with no bias for memory key / values
|
948
948
|
|
949
|
-
if has_mem_kv:
|
949
|
+
if exists(attn_bias) and has_mem_kv:
|
950
950
|
attn_bias = pad_at_dim(attn_bias, (self.num_mem_kv, 0), value = 0.)
|
951
951
|
|
952
952
|
# attention is all we need
|
@@ -4,11 +4,11 @@ x_transformers/autoregressive_wrapper.py,sha256=gYKIN5Rm8dMYSTX5yHpg9sPYyZf9rsRT
|
|
4
4
|
x_transformers/continuous.py,sha256=dpHK4NSMDQAJQ_N3Uj9rip0fYGXyu0QCCO_OfEdbRGs,6192
|
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=
|
7
|
+
x_transformers/x_transformers.py,sha256=F1ukKlTGhNaYmMLnm7nJy0XaUyAYwtrGGq8Gw94RPyQ,63919
|
8
8
|
x_transformers/xl_autoregressive_wrapper.py,sha256=DCx4n0_c1tFai4nOqaWVnqx2p9eutsZsDMiMP1ckxNU,4117
|
9
9
|
x_transformers/xval.py,sha256=EN3hxxleTRGYeAz6i4x3U_PrOm9TjxMF3eDhMKGx59E,8575
|
10
|
-
x_transformers-1.28.
|
11
|
-
x_transformers-1.28.
|
12
|
-
x_transformers-1.28.
|
13
|
-
x_transformers-1.28.
|
14
|
-
x_transformers-1.28.
|
10
|
+
x_transformers-1.28.1.dist-info/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
|
11
|
+
x_transformers-1.28.1.dist-info/METADATA,sha256=BjW_kBCR9LCWa1CbdAkd7vMBEVYkXTgLBbNw74Tz2R4,661
|
12
|
+
x_transformers-1.28.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
|
13
|
+
x_transformers-1.28.1.dist-info/top_level.txt,sha256=hO6KGpFuGucRNEtRfme4A_rGcM53AKwGP7RVlRIxS5Q,15
|
14
|
+
x_transformers-1.28.1.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|