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

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

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

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

@@ -91,7 +91,7 @@ class ViTEncoder(nn.Module):
         transformer_activation_kwargs: Optional[dict] = None,
         transformer_mlp_dropout=0.0,
         transformer_msa_dropout=0.1,
-        stochastic_depth=0.1,
+        transformer_stochastic_depth=0.1,
         linear_module=nn.Linear,
         initial_batch_norm=True,
     ):
@@ -287,7 +287,7 @@ class ViTEncoder(nn.Module):
                 activation_kwargs=transformer_activation_kwargs,
                 mlp_dropout=transformer_mlp_dropout,
                 msa_dropout=transformer_msa_dropout,
-                stochastic_depth=stochastic_depth,
+                stochastic_depth=transformer_stochastic_depth,
                 causal=False,
                 linear_module=linear_module,
                 bos_tokens=transformer_bos_tokens,
@@ -359,7 +359,7 @@ class CCT(nn.Module):
         transformer_activation_kwargs: Optional[dict] = None,
         transformer_mlp_dropout=0.0,
         transformer_msa_dropout=0.1,
-        stochastic_depth=0.1,
+        transformer_stochastic_depth=0.1,
         batch_norm_outputs=True,
         initial_batch_norm=True,
         linear_module=nn.Linear,
@@ -411,7 +411,7 @@ class CCT(nn.Module):
             transformer_activation_kwargs=transformer_activation_kwargs,
             mlp_dropout=transformer_mlp_dropout,
             msa_dropout=transformer_msa_dropout,
-            stochastic_depth=stochastic_depth,
+            transformer_stochastic_depth=transformer_stochastic_depth,
             linear_module=linear_module,
             initial_batch_norm=initial_batch_norm,
         )

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

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