broccoli-ml 0.24.2__py3-none-any.whl → 0.24.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.
broccoli/transformer.py CHANGED
@@ -264,7 +264,7 @@ class FeedforwardBlock(nn.Module):
264
264
  nn.LayerNorm(input_features),
265
265
  linear_module(input_features, self.max_features),
266
266
  self.activation,
267
- nn.LayerNorm(ratio * output_features) if raw_input else nn.Identity(),
267
+ # nn.LayerNorm(ratio * output_features) if raw_input else nn.Identity(),
268
268
  self.memory_type(ratio * output_features, output_features),
269
269
  self.dropout,
270
270
  ]
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: broccoli-ml
3
- Version: 0.24.2
3
+ Version: 0.24.3
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=4bxVDsO8E1d5-RZ23u160ZntazrT7Vt4AYTdAdCQU-w,3300
9
9
  broccoli/rope.py,sha256=hw7kBPNR9GQXj4GxyIAffsGKPfcTPOFh8Bc7oEHtaZY,12108
10
10
  broccoli/tensor.py,sha256=BVRS9-IWYtKXbXlerULQqitaUT8TPNgi7QTT5G_SmY4,3826
11
- broccoli/transformer.py,sha256=67y7FDfWAI-cIu8w5gRuG0MX24z1zflB218b1NKvk9A,16108
11
+ broccoli/transformer.py,sha256=2oq1dviGFehpjFSLFHFFhpxBYT-N094jD8owZXpeDv0,16110
12
12
  broccoli/utils.py,sha256=htq_hOsdhUhL0nJi9WkKiEYOjEoWqFpK5X49PtgTf-0,299
13
13
  broccoli/vit.py,sha256=pCjoJo6d1EhWmzI45sdP8t0olu6FlCGwMkGXcJJXXlo,15651
14
- broccoli_ml-0.24.2.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
15
- broccoli_ml-0.24.2.dist-info/METADATA,sha256=591Hk6OMRrWRzjyPS6To4t2uxOFa468XaPB9MlLSzZQ,1257
16
- broccoli_ml-0.24.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
17
- broccoli_ml-0.24.2.dist-info/RECORD,,
14
+ broccoli_ml-0.24.3.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
15
+ broccoli_ml-0.24.3.dist-info/METADATA,sha256=_VvJpa4Pcbx2VBmbzozogFz7O3OITJChjvEN3v3xYS8,1257
16
+ broccoli_ml-0.24.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
17
+ broccoli_ml-0.24.3.dist-info/RECORD,,