x-transformers 1.42.5__tar.gz → 1.42.6__tar.gz
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-1.42.5/x_transformers.egg-info → x_transformers-1.42.6}/PKG-INFO +1 -1
- {x_transformers-1.42.5 → x_transformers-1.42.6}/setup.py +1 -1
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/x_transformers.py +8 -2
- {x_transformers-1.42.5 → x_transformers-1.42.6/x_transformers.egg-info}/PKG-INFO +1 -1
- {x_transformers-1.42.5 → x_transformers-1.42.6}/LICENSE +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/README.md +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/setup.cfg +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/tests/test_x_transformers.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/__init__.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/attend.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/autoregressive_wrapper.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/continuous.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/dpo.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/multi_input.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/neo_mlp.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/nonautoregressive_wrapper.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/xl_autoregressive_wrapper.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers/xval.py +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers.egg-info/SOURCES.txt +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers.egg-info/dependency_links.txt +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers.egg-info/requires.txt +0 -0
- {x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers.egg-info/top_level.txt +0 -0
|
@@ -1055,7 +1055,12 @@ class Attention(Module):
|
|
|
1055
1055
|
logit_softclamp_value = 50.,
|
|
1056
1056
|
neutreno_value_residual = False, # Nguyen et al. https://arxiv.org/abs/2312.00751
|
|
1057
1057
|
neutreno_alpha = 0.4,
|
|
1058
|
-
onnxable = False
|
|
1058
|
+
onnxable = False,
|
|
1059
|
+
attend_sdp_kwargs: dict = dict(
|
|
1060
|
+
enable_flash = True,
|
|
1061
|
+
enable_math = True,
|
|
1062
|
+
enable_mem_efficient = True
|
|
1063
|
+
)
|
|
1059
1064
|
):
|
|
1060
1065
|
super().__init__()
|
|
1061
1066
|
dim_kv = default(dim_context, dim)
|
|
@@ -1188,7 +1193,8 @@ class Attention(Module):
|
|
|
1188
1193
|
softclamp_logits = softclamp_logits,
|
|
1189
1194
|
logit_softclamp_value = logit_softclamp_value,
|
|
1190
1195
|
cope = cope,
|
|
1191
|
-
onnxable = onnxable
|
|
1196
|
+
onnxable = onnxable,
|
|
1197
|
+
sdp_kwargs = attend_sdp_kwargs
|
|
1192
1198
|
)
|
|
1193
1199
|
|
|
1194
1200
|
# head scaling
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{x_transformers-1.42.5 → x_transformers-1.42.6}/x_transformers.egg-info/dependency_links.txt
RENAMED
|
File without changes
|
|
File without changes
|
|
File without changes
|