hyper-connections 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. hyper_connections/hyper_connections.py +2 -2
  2. {hyper_connections-0.0.3.dist-info → hyper_connections-0.0.4.dist-info}/METADATA +1 -1
  3. hyper_connections-0.0.4.dist-info/RECORD +6 -0
  4. hyper_connections-0.0.3.dist-info/RECORD +0 -6
  5. {hyper_connections-0.0.3.dist-info → hyper_connections-0.0.4.dist-info}/WHEEL +0 -0
  6. {hyper_connections-0.0.3.dist-info → hyper_connections-0.0.4.dist-info}/licenses/LICENSE +0 -0
hyper_connections/hyper_connections.py CHANGED
@@ -114,7 +114,7 @@ class HyperConnections(Module):
114
114
 
115
115
  return output
116
116
 
117
- def forward(self, residuals, **branch_kwargs):
117
+ def forward(self, residuals, *branch_args, **branch_kwargs):
118
118
 
119
119
  branch_input, residuals, beta = self.width_connection(residuals)
120
120
 
@@ -124,7 +124,7 @@ class HyperConnections(Module):
124
124
  if not exists(self.branch):
125
125
  return branch_input, add_residual_fn
126
126
 
127
- branch_output = self.branch(branch_input, **branch_kwargs)
127
+ branch_output = self.branch(branch_input, *branch_args, **branch_kwargs)
128
128
 
129
129
  (branch_output, *rest), tree_spec = tree_flatten(branch_output)
130
130
 
{hyper_connections-0.0.3.dist-info → hyper_connections-0.0.4.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: hyper-connections
3
- Version: 0.0.3
3
+ Version: 0.0.4
4
4
  Summary: Hyper-Connections
5
5
  Project-URL: Homepage, https://pypi.org/project/hyper-connections/
6
6
  Project-URL: Repository, https://github.com/lucidrains/hyper-connections
hyper_connections-0.0.4.dist-info/RECORD ADDED
@@ -0,0 +1,6 @@
1
+ hyper_connections/__init__.py,sha256=xXx2Mb-dS1__UPzT-5VR1XZmyqKSSkT1DU6bAcK8jR0,73
2
+ hyper_connections/hyper_connections.py,sha256=HbSNev1_wsJeZir5pqa-5IQYW3Ovy3UJD7A59utSF9Q,4331
3
+ hyper_connections-0.0.4.dist-info/METADATA,sha256=uRvNtEjISVXZkI_GVfwEk_1z_w2C4uzFYSP9m8cYpWs,4676
4
+ hyper_connections-0.0.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
5
+ hyper_connections-0.0.4.dist-info/licenses/LICENSE,sha256=E7RGS7kpJIStk5za_-4DVhWEAamf65EU0CNML25mq4c,1066
6
+ hyper_connections-0.0.4.dist-info/RECORD,,
hyper_connections-0.0.3.dist-info/RECORD DELETED
@@ -1,6 +0,0 @@
1
- hyper_connections/__init__.py,sha256=xXx2Mb-dS1__UPzT-5VR1XZmyqKSSkT1DU6bAcK8jR0,73
2
- hyper_connections/hyper_connections.py,sha256=91QtTtnpffmErIZvrnTtosSf4JgBqcyGvxftmka-EOw,4303
3
- hyper_connections-0.0.3.dist-info/METADATA,sha256=8XKTmC6Ys10uOyotPtvL17v4uZyepkWoeMRVM4B_TSQ,4676
4
- hyper_connections-0.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
5
- hyper_connections-0.0.3.dist-info/licenses/LICENSE,sha256=E7RGS7kpJIStk5za_-4DVhWEAamf65EU0CNML25mq4c,1066
6
- hyper_connections-0.0.3.dist-info/RECORD,,