titans-pytorch 0.0.17__tar.gz → 0.0.18__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.17 → titans_pytorch-0.0.18}/PKG-INFO +1 -1
  2. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/pyproject.toml +1 -1
  3. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/titans_pytorch/titans.py +1 -1
  4. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/train.py +0 -1
  5. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/.github/workflows/python-publish.yml +0 -0
  6. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/.github/workflows/test.yaml +0 -0
  7. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/.gitignore +0 -0
  8. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/LICENSE +0 -0
  9. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/README.md +0 -0
  10. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/data/README.md +0 -0
  11. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/data/enwik8.gz +0 -0
  12. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/fig1.png +0 -0
  13. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/fig2.png +0 -0
  14. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/requirements.txt +0 -0
  15. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/tests/test_titans.py +0 -0
  16. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/titans_pytorch/__init__.py +0 -0
  17. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/titans_pytorch/associative_scan.py +0 -0
  18. {titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/titans_pytorch/titans_attn_memory.py +0 -0
{titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: titans-pytorch
3
- Version: 0.0.17
3
+ Version: 0.0.18
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.17 → titans_pytorch-0.0.18}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "titans-pytorch"
3
- version = "0.0.17"
3
+ version = "0.0.18"
4
4
  description = "Titans"
5
5
  authors = [
6
6
  { name = "Phil Wang", email = "lucidrains@gmail.com" }
{titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/titans_pytorch/titans.py RENAMED
@@ -269,7 +269,7 @@ class NeuralMemory(Module):
269
269
  gates = F.pad(gates, (0, next_power_two_seq_len - seq_len))
270
270
  inputs = F.pad(inputs, (0, next_power_two_seq_len - seq_len))
271
271
 
272
- outputs = scan(gates, inputs)
272
+ outputs = scan(gates.contiguous(), inputs.contiguous())
273
273
 
274
274
  outputs = outputs[..., :seq_len]
275
275
  outputs = rearrange(outputs, 'b d n -> b n d')
{titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/train.py RENAMED
@@ -15,7 +15,6 @@ from taylor_series_linear_attention import TaylorSeriesLinearAttn
15
15
 
16
16
  from titans_pytorch.titans import (
17
17
  NeuralMemory,
18
- MemoryAttention,
19
18
  MemoryMLP
20
19
  )
21
20
 
{titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/LICENSE RENAMED
File without changes