broccoli-ml 13.0.0__tar.gz → 13.0.1__tar.gz

{broccoli_ml-13.0.0 → broccoli_ml-13.0.1}/PKG-INFO

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

{broccoli_ml-13.0.0 → broccoli_ml-13.0.1}/broccoli/vit.py

@@ -187,9 +187,14 @@ class ViTEncoder(nn.Module):
         transformer_stochastic_depth=0.1,
         transformer_checkpoint_ff=True,
         linear_module=nn.Linear,
+        alpha=1.0,
+        beta=1.0,
     ):
         super().__init__()
 
+        self.alpha = alpha
+        self.beta = beta
+
         if cnn_activation_kwargs is not None:
             self.cnn_activation = cnn_activation(**cnn_activation_kwargs)
         else:
@@ -491,6 +496,8 @@ class ViT(nn.Module):
         batch_norm_logits=True,
         logit_projection_layer=nn.Linear,
         linear_module=nn.Linear,
+        alpha=1.0,
+        beta=1.0,
     ):
 
         super().__init__()
@@ -511,6 +518,9 @@ class ViT(nn.Module):
                 "SwiGLU": SwiGLU,
             }[transformer_activation]
 
+        self.alpha = alpha
+        self.beta = beta
+
         self.encoder = ViTEncoder(
             input_size=input_size,
             initial_batch_norm=initial_batch_norm,
@@ -559,6 +569,8 @@ class ViT(nn.Module):
             transformer_stochastic_depth=transformer_stochastic_depth,
             transformer_checkpoint_ff=transformer_checkpoint_ff,
             linear_module=linear_module,
+            alpha=alpha,
+            beta=beta,
         )
 
         self.pool = head(

{broccoli_ml-13.0.0 → broccoli_ml-13.0.1}/pyproject.toml

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