broccoli-ml 0.5.2__tar.gz → 0.5.3__tar.gz

{broccoli_ml-0.5.2 → broccoli_ml-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: broccoli-ml
-Version: 0.5.2
+Version: 0.5.3
 Summary: Some useful Pytorch models, circa 2025
 License: MIT
 Author: Nicholas Bailey

{broccoli_ml-0.5.2 → broccoli_ml-0.5.3}/broccoli/vit.py

@@ -285,8 +285,8 @@ class ViTEncoder(nn.Module):
                 mlp_ratio=transformer_mlp_ratio,
                 activation=transformer_activation,
                 activation_kwargs=transformer_activation_kwargs,
-                mlp_dropout=transformer_mlp_dropout,
-                msa_dropout=transformer_msa_dropout,
+                transformer_mlp_dropout=transformer_mlp_dropout,
+                transformer_msa_dropout=transformer_msa_dropout,
                 stochastic_depth=transformer_stochastic_depth,
                 causal=False,
                 linear_module=linear_module,
@@ -409,8 +409,8 @@ class CCT(nn.Module):
             transformer_bos_tokens=transformer_bos_tokens,
             transformer_activation=transformer_activation,
             transformer_activation_kwargs=transformer_activation_kwargs,
-            mlp_dropout=transformer_mlp_dropout,
-            msa_dropout=transformer_msa_dropout,
+            transformer_mlp_dropout=transformer_mlp_dropout,
+            transformer_msa_dropout=transformer_msa_dropout,
             transformer_stochastic_depth=transformer_stochastic_depth,
             linear_module=linear_module,
             initial_batch_norm=initial_batch_norm,

{broccoli_ml-0.5.2 → broccoli_ml-0.5.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "broccoli-ml"
-version = "0.5.2"
+version = "0.5.3"
 description = "Some useful Pytorch models, circa 2025"
 authors = [
     {name = "Nicholas Bailey"}