broccoli-ml 0.21.0__py3-none-any.whl → 0.21.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.
broccoli/transformer.py CHANGED
@@ -262,14 +262,14 @@ class FeedforwardBlock(nn.Module):
262
262
  *[
263
263
  (
264
264
  nn.LayerNorm(input_features)
265
- if not self.regularise_values
265
+ if not regularise_values
266
266
  else nn.Identity()
267
267
  ),
268
268
  linear_module(input_features, self.max_features),
269
269
  self.activation,
270
270
  (
271
271
  nn.LayerNorm(input_features)
272
- if not self.regularise_values
272
+ if not regularise_values
273
273
  else nn.Identity()
274
274
  ),
275
275
  self.memory_type(ratio * output_features, output_features),
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: broccoli-ml
3
- Version: 0.21.0
3
+ Version: 0.21.1
4
4
  Summary: Some useful Pytorch models, circa 2025
5
5
  License: MIT
6
6
  Author: Nicholas Bailey
@@ -8,10 +8,10 @@ broccoli/eigenpatches.py,sha256=J6n2usN1oQuHEHYiBNyYpn_a9eQcHjOBiIlvSei520Y,2413
8
8
  broccoli/linear.py,sha256=g8YrxNl6g_WcHrWVmbaBHJU5hv6daFS0r4TxAoPJ9UE,3012
9
9
  broccoli/rope.py,sha256=hw7kBPNR9GQXj4GxyIAffsGKPfcTPOFh8Bc7oEHtaZY,12108
10
10
  broccoli/tensor.py,sha256=MUvXtwD2f1sPTBym4FB0x_ZfsJUBNLgULUlN8btV8GI,1943
11
- broccoli/transformer.py,sha256=2O9EALlZPQXYCtOGvkCgoWv0-K8rNtsp-MTon1h_4n8,16343
11
+ broccoli/transformer.py,sha256=mFmQiuOu94yKVPZVsd0Cv5FGqkGXOX6zkGAQLqjThKg,16333
12
12
  broccoli/utils.py,sha256=htq_hOsdhUhL0nJi9WkKiEYOjEoWqFpK5X49PtgTf-0,299
13
13
  broccoli/vit.py,sha256=vQtOcC0Dd8y6PTWx0xCbnE4ymYkL_HfYrerqaJ0hs1k,16404
14
- broccoli_ml-0.21.0.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
15
- broccoli_ml-0.21.0.dist-info/METADATA,sha256=jtoA3JoVE-R_VYlofiCzNxnWtzvV1Tu1HjR_Bm9T8lA,1257
16
- broccoli_ml-0.21.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
17
- broccoli_ml-0.21.0.dist-info/RECORD,,
14
+ broccoli_ml-0.21.1.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
15
+ broccoli_ml-0.21.1.dist-info/METADATA,sha256=ZhxdgQ_AVABg6P-rzyssN3j62lU21YWIgnphuf7v2o0,1257
16
+ broccoli_ml-0.21.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
17
+ broccoli_ml-0.21.1.dist-info/RECORD,,