hyper-connections 0.0.14__py3-none-any.whl → 0.0.15__py3-none-any.whl

hyper_connections/hyper_connections.py

@@ -121,7 +121,11 @@ class HyperConnections(Module):
         self.channel_first = channel_first
 
     @classmethod
-    def get_expand_reduce_stream_functions(cls, num_streams):
+    def get_expand_reduce_stream_functions(cls, num_streams, disable = False):
+
+        if disable:
+            return (identity, identity)
+
         expand_fn = partial(repeat, pattern = 'b ... -> (b s) ...', s = num_streams)
         reduce_fn = partial(reduce, pattern = '(b s) ... -> b ...', reduction = 'sum', s = num_streams)
 
@@ -133,7 +137,7 @@ class HyperConnections(Module):
         hyper_conn_klass = cls if not disable else Residual
 
         init_hyper_conn_fn = partial(hyper_conn_klass, num_streams)
-        expand_reduce_fns = cls.get_expand_reduce_stream_functions(num_streams) if not disable else (identity, identity)
+        expand_reduce_fns = cls.get_expand_reduce_stream_functions(num_streams, disable = disable)
 
         return (init_hyper_conn_fn, *expand_reduce_fns)
 

hyper_connections/hyper_connections_with_multi_branch_inputs.py

@@ -77,6 +77,9 @@ class HyperConnections(Module):
 
     @classmethod
     def get_expand_reduce_stream_functions(cls, num_streams):
+        if disable:
+            return (identity, identity)
+
         expand_fn = partial(repeat, pattern = 'b ... -> (b s) ...', s = num_streams)
         reduce_fn = partial(reduce, pattern = '(b s) ... -> b ...', reduction = 'sum', s = num_streams)
 
@@ -88,7 +91,7 @@ class HyperConnections(Module):
         hyper_conn_klass = cls if not disable else Residual
 
         init_hyper_conn_fn = partial(hyper_conn_klass, num_streams)
-        expand_reduce_fns = cls.get_expand_reduce_stream_functions(num_streams) if not disable else (identity, identity)
+        expand_reduce_fns = cls.get_expand_reduce_stream_functions(num_streams, disable = disable)
 
         return (init_hyper_conn_fn, *expand_reduce_fns)
 

{hyper_connections-0.0.14.dist-info → hyper_connections-0.0.15.dist-info}/METADATA

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

hyper_connections-0.0.15.dist-info/RECORD

@@ -0,0 +1,7 @@
+hyper_connections/__init__.py,sha256=mAy66IuHqXM4XOyOZGt5mo2B2hfHdUk8jW31YnWNQTg,104
+hyper_connections/hyper_connections.py,sha256=LoyiGGDH81MLUcF9WKT39lA9Y1CXUAd-SYM9tJGQ61A,8099
+hyper_connections/hyper_connections_with_multi_branch_inputs.py,sha256=6KHdqvjUZCyHwf7LVByMMuoQtnEHBxRxdt-4Bw9HPcA,6130
+hyper_connections-0.0.15.dist-info/METADATA,sha256=DnZongXit3VTaT7DxmpK0RRfUhXqfUTjM32a-ofbrIw,5076
+hyper_connections-0.0.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyper_connections-0.0.15.dist-info/licenses/LICENSE,sha256=E7RGS7kpJIStk5za_-4DVhWEAamf65EU0CNML25mq4c,1066
+hyper_connections-0.0.15.dist-info/RECORD,,

hyper_connections-0.0.14.dist-info/RECORD

@@ -1,7 +0,0 @@
-hyper_connections/__init__.py,sha256=mAy66IuHqXM4XOyOZGt5mo2B2hfHdUk8jW31YnWNQTg,104
-hyper_connections/hyper_connections.py,sha256=-cqKAGkvSf9NKP-S5cmZU9h59qyPbDq27CSDTmZ5fm8,8042
-hyper_connections/hyper_connections_with_multi_branch_inputs.py,sha256=jn-wxAgECVUpiyf1elTtwcr-wvzCQQVS32rxted7XqU,6091
-hyper_connections-0.0.14.dist-info/METADATA,sha256=mM6YzyPTNDYZqR-l8cwqxnZfYngaNf0LunycWmIoP4Y,5076
-hyper_connections-0.0.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hyper_connections-0.0.14.dist-info/licenses/LICENSE,sha256=E7RGS7kpJIStk5za_-4DVhWEAamf65EU0CNML25mq4c,1066
-hyper_connections-0.0.14.dist-info/RECORD,,