broccoli-ml 0.9.0__py3-none-any.whl → 0.10.0__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.
Files changed (6) hide show
  1. broccoli/transformer.py +1 -2
  2. broccoli/vit.py +0 -1
  3. {broccoli_ml-0.9.0.dist-info → broccoli_ml-0.10.0.dist-info}/METADATA +1 -1
  4. {broccoli_ml-0.9.0.dist-info → broccoli_ml-0.10.0.dist-info}/RECORD +6 -6
  5. {broccoli_ml-0.9.0.dist-info → broccoli_ml-0.10.0.dist-info}/LICENSE +0 -0
  6. {broccoli_ml-0.9.0.dist-info → broccoli_ml-0.10.0.dist-info}/WHEEL +0 -0
broccoli/transformer.py CHANGED
@@ -233,7 +233,6 @@ class FeedforwardLayer(nn.Module):
233
233
  output_features,
234
234
  activation=nn.ReLU,
235
235
  activation_kwargs=None,
236
- layer_norm=True,
237
236
  dropout=0.0,
238
237
  linear_module=nn.Linear,
239
238
  ):
@@ -248,7 +247,7 @@ class FeedforwardLayer(nn.Module):
248
247
 
249
248
  self.process = nn.Sequential(
250
249
  *[
251
- nn.LayerNorm(input_features) if layer_norm else nn.Identity(),
250
+ nn.LayerNorm(input_features),
252
251
  linear_module(
253
252
  input_features,
254
253
  (
broccoli/vit.py CHANGED
@@ -261,7 +261,6 @@ class ViTEncoder(nn.Module):
261
261
  transformer_embedding_size,
262
262
  activation=transformer_activation,
263
263
  activation_kwargs=transformer_activation_kwargs,
264
- layer_norm=False, # rely on initial batch_norm
265
264
  dropout=transformer_mlp_dropout,
266
265
  linear_module=linear_module,
267
266
  )
{broccoli_ml-0.9.0.dist-info → broccoli_ml-0.10.0.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: broccoli-ml
3
- Version: 0.9.0
3
+ Version: 0.10.0
4
4
  Summary: Some useful Pytorch models, circa 2025
5
5
  License: MIT
6
6
  Author: Nicholas Bailey
{broccoli_ml-0.9.0.dist-info → broccoli_ml-0.10.0.dist-info}/RECORD RENAMED
@@ -8,10 +8,10 @@ broccoli/eigenpatches.py,sha256=J6n2usN1oQuHEHYiBNyYpn_a9eQcHjOBiIlvSei520Y,2413
8
8
  broccoli/linear.py,sha256=0XYCi3ckTEKwAgBOMUSJP2HsnrroOH8eyrhRdpANG2w,1298
9
9
  broccoli/rope.py,sha256=hw7kBPNR9GQXj4GxyIAffsGKPfcTPOFh8Bc7oEHtaZY,12108
10
10
  broccoli/tensor.py,sha256=E2JK5mQwJf75e23-JGcDoT7QxQf89DJReUo2et1LhRY,1716
11
- broccoli/transformer.py,sha256=rxYImDa2FzFe671ZYsGIwpW4KAps6wICq3KC7Pel9f0,15988
11
+ broccoli/transformer.py,sha256=RSZpbHs_K4ts5os6lWxcGDI3p0zreRwQNnk6mV8HJnk,15930
12
12
  broccoli/utils.py,sha256=htq_hOsdhUhL0nJi9WkKiEYOjEoWqFpK5X49PtgTf-0,299
13
- broccoli/vit.py,sha256=PZMHf73aomw5mc3EjIpwNlz_TplgIJqduEmdjtc5giM,15249
14
- broccoli_ml-0.9.0.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
15
- broccoli_ml-0.9.0.dist-info/METADATA,sha256=NdOXxGC-7qZ_eprB8vBvLYmhu69gRkj60LIJs4wZlUY,1256
16
- broccoli_ml-0.9.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
17
- broccoli_ml-0.9.0.dist-info/RECORD,,
13
+ broccoli/vit.py,sha256=uXqMIvAVY4PuA-Fv1YxU7L3_74fR19GtNu1caQeMr6k,15185
14
+ broccoli_ml-0.10.0.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
15
+ broccoli_ml-0.10.0.dist-info/METADATA,sha256=zRP8Hn-Q0uMDvNhInEuwcm-sVwG5zE3vsWZBSuVqDmE,1257
16
+ broccoli_ml-0.10.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
17
+ broccoli_ml-0.10.0.dist-info/RECORD,,