broccoli-ml 9.2.0__py3-none-any.whl → 9.2.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.
Files changed (5) hide show
  1. broccoli/transformer.py +6 -2
  2. {broccoli_ml-9.2.0.dist-info → broccoli_ml-9.2.1.dist-info}/METADATA +1 -1
  3. {broccoli_ml-9.2.0.dist-info → broccoli_ml-9.2.1.dist-info}/RECORD +5 -5
  4. {broccoli_ml-9.2.0.dist-info → broccoli_ml-9.2.1.dist-info}/LICENSE +0 -0
  5. {broccoli_ml-9.2.0.dist-info → broccoli_ml-9.2.1.dist-info}/WHEEL +0 -0
broccoli/transformer.py CHANGED
@@ -391,8 +391,12 @@ class FeedforwardBlock(nn.Module):
391
391
  self.master_recycling_rate = self.linear_in.row_recycling_rate
392
392
  self.linear_in.row_recycling_rate = 0.0
393
393
  self.linear_out.column_recycling_rate = 0.0
394
- if hasattr(self.linear_in, "column_recycling_rate") or hasattr(
395
- self.linear_out, "row_recycling_rate"
394
+ if (
395
+ hasattr(self.linear_in, "column_recycling_rate")
396
+ and self.linear_in.column_recycling_rate > 0
397
+ ) or (
398
+ hasattr(self.linear_out, "row_recycling_rate")
399
+ and self.linear_out.row_recycling_rate > 0
396
400
  ):
397
401
  raise NotImplementedError(
398
402
  "At the moment this layer can only support recycling linear "
{broccoli_ml-9.2.0.dist-info → broccoli_ml-9.2.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: broccoli-ml
3
- Version: 9.2.0
3
+ Version: 9.2.1
4
4
  Summary: Some useful Pytorch models, circa 2025
5
5
  License: MIT
6
6
  Author: Nicholas Bailey
{broccoli_ml-9.2.0.dist-info → broccoli_ml-9.2.1.dist-info}/RECORD RENAMED
@@ -4,10 +4,10 @@ broccoli/cnn.py,sha256=WjoPDSpe3ttwxCBNfCVRdaCHvbeZ7G-a5_i8fUsK_d8,4889
4
4
  broccoli/linear.py,sha256=7uN7zVPJ6Ptec31O8a-GvWT5nZk56Wf1RLJRvUAT0yo,11406
5
5
  broccoli/rope.py,sha256=GRqApBNmYCFaDak0WL1xE_BC5CTTYKQU_PBdeTcQcjc,12557
6
6
  broccoli/tensor.py,sha256=um8mrxkYbvNDo-QvHlmJm8Aw6qcngOlUZPoAk_PMReA,4480
7
- broccoli/transformer.py,sha256=ckwFpNTAeYB_V8F-_DwT0Z5--QUEepM4r6xlSxhCY68,26079
7
+ broccoli/transformer.py,sha256=r-ggAeNDW5QpBi9As1U9sIfxITBOx0WHk_K4zWpyTM8,26233
8
8
  broccoli/utils.py,sha256=oOWzn6dJ5nC_9r4zq0emmfmaYACJXJNFS48AOpW2jqc,358
9
9
  broccoli/vit.py,sha256=sC6K3FK3a8ojOgvNWSWhuZHBtnFrrTQbsDdlagcKJH4,22224
10
- broccoli_ml-9.2.0.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
11
- broccoli_ml-9.2.0.dist-info/METADATA,sha256=MnrReG_teIm7YceDS9nfvXnelEXHGP4JFhoz9dhkJWo,1368
12
- broccoli_ml-9.2.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
13
- broccoli_ml-9.2.0.dist-info/RECORD,,
10
+ broccoli_ml-9.2.1.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
11
+ broccoli_ml-9.2.1.dist-info/METADATA,sha256=Nj7WnXKxlvSlrK8rQp9wizgPGs7ZMnhCi-KY5O6W-wc,1368
12
+ broccoli_ml-9.2.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
13
+ broccoli_ml-9.2.1.dist-info/RECORD,,