hyper-connections 0.1.1__tar.gz → 0.1.4__tar.gz

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 (13) hide show
  1. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/PKG-INFO +1 -1
  2. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/hyper_connections/hyper_connections.py +0 -4
  3. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/pyproject.toml +1 -1
  4. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/.github/workflows/python-publish.yml +0 -0
  5. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/.github/workflows/test.yml +0 -0
  6. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/.gitignore +0 -0
  7. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/LICENSE +0 -0
  8. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/README.md +0 -0
  9. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/hyper-connections.png +0 -0
  10. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/hyper_connections/__init__.py +0 -0
  11. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/hyper_connections/hyper_connections_with_multi_branch_inputs.py +0 -0
  12. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/hyper_connections/hyper_connections_with_multi_input_streams.py +0 -0
  13. {hyper_connections-0.1.1 → hyper_connections-0.1.4}/tests/test_hyper_connections.py +0 -0
{hyper_connections-0.1.1 → hyper_connections-0.1.4}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: hyper-connections
3
- Version: 0.1.1
3
+ Version: 0.1.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.1.1 → hyper_connections-0.1.4}/hyper_connections/hyper_connections.py RENAMED
@@ -12,8 +12,6 @@ from torch.utils._pytree import tree_flatten, tree_unflatten
12
12
 
13
13
  from einops import rearrange, repeat, reduce, einsum
14
14
 
15
- from beartype import beartype
16
-
17
15
  """
18
16
  ein notation:
19
17
  b - batch
@@ -70,7 +68,6 @@ class RMSNorm(Module):
70
68
  # residual base class
71
69
 
72
70
  class Residual(Module):
73
- @beartype
74
71
  def __init__(
75
72
  self,
76
73
  *args,
@@ -121,7 +118,6 @@ class Residual(Module):
121
118
  # hyper connection residual streams
122
119
 
123
120
  class HyperConnections(Module):
124
- @beartype
125
121
  def __init__(
126
122
  self,
127
123
  num_residual_streams,
{hyper_connections-0.1.1 → hyper_connections-0.1.4}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "hyper-connections"
3
- version = "0.1.1"
3
+ version = "0.1.4"
4
4
  description = "Hyper-Connections"
5
5
  authors = [
6
6
  { name = "Phil Wang", email = "lucidrains@gmail.com" }