titans-pytorch 0.2.19__py3-none-any.whl → 0.2.20__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.
- titans_pytorch/mac_transformer.py +1 -1
- {titans_pytorch-0.2.19.dist-info → titans_pytorch-0.2.20.dist-info}/METADATA +2 -2
- {titans_pytorch-0.2.19.dist-info → titans_pytorch-0.2.20.dist-info}/RECORD +5 -5
- {titans_pytorch-0.2.19.dist-info → titans_pytorch-0.2.20.dist-info}/WHEEL +0 -0
- {titans_pytorch-0.2.19.dist-info → titans_pytorch-0.2.20.dist-info}/licenses/LICENSE +0 -0
|
@@ -560,7 +560,7 @@ class MemoryAsContextTransformer(Module):
|
|
|
560
560
|
chunk_size = self.neural_memory_segment_len,
|
|
561
561
|
batch_size = neural_memory_batch_size,
|
|
562
562
|
model = deepcopy(neural_memory_model),
|
|
563
|
-
|
|
563
|
+
accept_weight_residual = neural_mem_weight_residual and not is_first_neural_mem,
|
|
564
564
|
**neural_memory_kwargs
|
|
565
565
|
)
|
|
566
566
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: titans-pytorch
|
|
3
|
-
Version: 0.2.
|
|
3
|
+
Version: 0.2.20
|
|
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
|
|
@@ -204,6 +204,6 @@ $ python train_mac.py
|
|
|
204
204
|
eprint = {2501.12352},
|
|
205
205
|
archivePrefix = {arXiv},
|
|
206
206
|
primaryClass = {cs.LG},
|
|
207
|
-
url
|
|
207
|
+
url = {https://arxiv.org/abs/2501.12352},
|
|
208
208
|
}
|
|
209
209
|
```
|
|
@@ -1,9 +1,9 @@
|
|
|
1
1
|
titans_pytorch/__init__.py,sha256=Y3m_ZlpEqYwp-Md1ARhNGJxq8bQp8ty1o039nZOOJo0,276
|
|
2
2
|
titans_pytorch/associative_scan.py,sha256=Y-iYqmFuG-NoCKu6kgql1mhowXTeJfyawi3eUIXamp0,2650
|
|
3
|
-
titans_pytorch/mac_transformer.py,sha256=
|
|
3
|
+
titans_pytorch/mac_transformer.py,sha256=5rO4GQxSyFWWEc3pc3xNyG0sK5EXE7MmxKI-_kEMl2M,24941
|
|
4
4
|
titans_pytorch/memory_models.py,sha256=Q9SAIyAbStF5Tz0EhvRbn3yAdE3nk3xKc1ndieIe714,4671
|
|
5
5
|
titans_pytorch/neural_memory.py,sha256=e1TENsK0IptpapFRtQux-Uii2MFvisoiQLzTQboHc50,26709
|
|
6
|
-
titans_pytorch-0.2.
|
|
7
|
-
titans_pytorch-0.2.
|
|
8
|
-
titans_pytorch-0.2.
|
|
9
|
-
titans_pytorch-0.2.
|
|
6
|
+
titans_pytorch-0.2.20.dist-info/METADATA,sha256=8v5puM3CwbPaYDt2aZaqWruPj8whCt_moiQUVE1vk5U,6816
|
|
7
|
+
titans_pytorch-0.2.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
|
8
|
+
titans_pytorch-0.2.20.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
|
|
9
|
+
titans_pytorch-0.2.20.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|