broccoli-ml 0.8.0__py3-none-any.whl → 0.8.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/vit.py CHANGED
@@ -312,7 +312,7 @@ class CCT(nn.Module):
312
312
  input_size=(32, 32),
313
313
  cnn=True,
314
314
  cnn_in_channels=3,
315
- minimum_cnn_out_channels=16,
315
+ cnn_out_channels=16,
316
316
  cnn_kernel_size=3,
317
317
  cnn_kernel_stride=1,
318
318
  cnn_padding="same",
@@ -365,7 +365,7 @@ class CCT(nn.Module):
365
365
  input_size=input_size,
366
366
  cnn=cnn,
367
367
  cnn_in_channels=cnn_in_channels,
368
- minimum_cnn_out_channels=minimum_cnn_out_channels,
368
+ cnn_out_channels=cnn_out_channels,
369
369
  cnn_kernel_size=cnn_kernel_size,
370
370
  cnn_kernel_stride=cnn_kernel_stride,
371
371
  cnn_padding=cnn_padding,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: broccoli-ml
3
- Version: 0.8.0
3
+ Version: 0.8.1
4
4
  Summary: Some useful Pytorch models, circa 2025
5
5
  License: MIT
6
6
  Author: Nicholas Bailey
@@ -10,8 +10,8 @@ broccoli/rope.py,sha256=hw7kBPNR9GQXj4GxyIAffsGKPfcTPOFh8Bc7oEHtaZY,12108
10
10
  broccoli/tensor.py,sha256=E2JK5mQwJf75e23-JGcDoT7QxQf89DJReUo2et1LhRY,1716
11
11
  broccoli/transformer.py,sha256=RSZpbHs_K4ts5os6lWxcGDI3p0zreRwQNnk6mV8HJnk,15930
12
12
  broccoli/utils.py,sha256=htq_hOsdhUhL0nJi9WkKiEYOjEoWqFpK5X49PtgTf-0,299
13
- broccoli/vit.py,sha256=qNavAe_jNlslYlLsXmScHTnLuL3-MAVCAhMsJt3v5Rg,15209
14
- broccoli_ml-0.8.0.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
15
- broccoli_ml-0.8.0.dist-info/METADATA,sha256=LVv2HQIjuvpHEH9uixMzbdE-VeJ44E2TizerorwHYS0,1256
16
- broccoli_ml-0.8.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
17
- broccoli_ml-0.8.0.dist-info/RECORD,,
13
+ broccoli/vit.py,sha256=uXqMIvAVY4PuA-Fv1YxU7L3_74fR19GtNu1caQeMr6k,15185
14
+ broccoli_ml-0.8.1.dist-info/LICENSE,sha256=0BAzJE5BqQ7Iixp_AFdB2W1uO-HCRX-Qfun8PHt6yVM,1073
15
+ broccoli_ml-0.8.1.dist-info/METADATA,sha256=ofw4KzrRqP9e5i5OHlu-zxSpMO2DaohGA_N3RfAmt7s,1256
16
+ broccoli_ml-0.8.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
17
+ broccoli_ml-0.8.1.dist-info/RECORD,,