hyper-connections 0.3.0__tar.gz → 0.3.2__tar.gz

{hyper_connections-0.3.0 → hyper_connections-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyper-connections
-Version: 0.3.0
+Version: 0.3.2
 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.2/hyper_connections/__init__.py

@@ -0,0 +1,18 @@
+from hyper_connections.hyper_connections import (
+    HyperConnections,
+    get_expand_reduce_stream_functions,
+    get_init_and_expand_reduce_stream_functions,
+    Residual,
+    StreamEmbed,
+    AttentionPoolReduceStream
+)
+
+# export with mc prefix, as well as mHC
+
+from hyper_connections.manifold_constrained_hyper_connections import (
+    ManifoldConstrainedHyperConnections,
+    get_expand_reduce_stream_functions as mc_get_expand_reduce_stream_functions,
+    get_init_and_expand_reduce_stream_functions as mc_get_init_and_expand_reduce_stream_functions
+)
+
+mHC = ManifoldConstrainedHyperConnections

{hyper_connections-0.3.0 → hyper_connections-0.3.2}/hyper_connections/manifold_constrained_hyper_connections.py

@@ -54,6 +54,8 @@ def l1norm(t, dim):
     return F.normalize(t, p = 1, dim = dim)
 
 def sinkhorn_knopps(log_alpha, iters = 20):
+    log_alpha = log_alpha - log_alpha.amax(dim = -2, keepdim = True)
+
     alpha = log_alpha.exp()
 
     for _ in range(iters):

{hyper_connections-0.3.0 → hyper_connections-0.3.2}/pyproject.toml

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

hyper_connections-0.3.0/hyper_connections/__init__.py

@@ -1,8 +0,0 @@
-from hyper_connections.hyper_connections import (
-    HyperConnections,
-    get_expand_reduce_stream_functions,
-    get_init_and_expand_reduce_stream_functions,
-    Residual,
-    StreamEmbed,
-    AttentionPoolReduceStream
-)