hyper-connections 0.3.7__tar.gz → 0.3.8__tar.gz

{hyper_connections-0.3.7 → hyper_connections-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyper-connections
-Version: 0.3.7
+Version: 0.3.8
 Summary: Hyper-Connections
 Project-URL: Homepage, https://pypi.org/project/hyper-connections/
 Project-URL: Repository, https://github.com/lucidrains/hyper-connections

{hyper_connections-0.3.7 → hyper_connections-0.3.8}/hyper_connections/manifold_constrained_hyper_connections.py

@@ -200,7 +200,8 @@ class ManifoldConstrainedHyperConnections(Module):
         num_input_views = 1,                # allow for the branch module to receive multiple input views, dimension placed on the very left (before batch)
         depth_residual_fn = add,
         num_fracs = 1,                      # https://arxiv.org/abs/2503.14125
-        sinkhorn_iters = 20
+        sinkhorn_iters = 20,
+        forward_method_names: tuple[str, ...] = (),
     ):
         """
         Appendix J, Algorithm2 in - https://arxiv.org/abs/2409.19606
@@ -290,6 +291,16 @@ class ManifoldConstrainedHyperConnections(Module):
 
         self.depth_residual_fn = depth_residual_fn
 
+        # forwarding method names
+
+        self.forward_method_names = forward_method_names
+
+        for forward_method_name in self.forward_method_names:
+            assert not hasattr(self, forward_method_name)
+
+            fn = getattr(self.branch, forward_method_name)
+            setattr(self, forward_method_name, fn)
+
     def width_connection(
         self,
         residuals

{hyper_connections-0.3.7 → hyper_connections-0.3.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hyper-connections"
-version = "0.3.7"
+version = "0.3.8"
 description = "Hyper-Connections"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }