titans-pytorch 0.0.24__tar.gz → 0.0.25__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 (18) hide show
  1. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/PKG-INFO +1 -1
  2. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/pyproject.toml +1 -1
  3. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/.github/workflows/python-publish.yml +0 -0
  4. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/.github/workflows/test.yaml +0 -0
  5. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/.gitignore +0 -0
  6. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/LICENSE +0 -0
  7. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/README.md +0 -0
  8. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/data/README.md +0 -0
  9. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/data/enwik8.gz +0 -0
  10. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/fig1.png +0 -0
  11. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/fig2.png +0 -0
  12. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/requirements.txt +0 -0
  13. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/tests/test_titans.py +0 -0
  14. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/titans_pytorch/__init__.py +0 -0
  15. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/titans_pytorch/associative_scan.py +0 -0
  16. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/titans_pytorch/titans.py +1 -1
  17. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/titans_pytorch/titans_attn_memory.py +0 -0
  18. {titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/train.py +0 -0
{titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: titans-pytorch
3
- Version: 0.0.24
3
+ Version: 0.0.25
4
4
  Summary: Titans
5
5
  Project-URL: Homepage, https://pypi.org/project/titans-pytorch/
6
6
  Project-URL: Repository, https://github.com/lucidrains/titans-pytorch
{titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "titans-pytorch"
3
- version = "0.0.24"
3
+ version = "0.0.25"
4
4
  description = "Titans"
5
5
  authors = [
6
6
  { name = "Phil Wang", email = "lucidrains@gmail.com" }
{titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/LICENSE RENAMED
File without changes
{titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/titans_pytorch/titans.py RENAMED
@@ -136,6 +136,7 @@ class NeuralMemory(Module):
136
136
  )
137
137
  ):
138
138
  super().__init__()
139
+ dim_head = default(dim_head, dim)
139
140
 
140
141
  # norms
141
142
 
@@ -146,7 +147,6 @@ class NeuralMemory(Module):
146
147
 
147
148
  # maybe multi-headed
148
149
 
149
- dim_head = default(dim_head, dim)
150
150
  dim_inner = dim_head * heads
151
151
 
152
152
  self.heads = heads