titans-pytorch 0.2.11__py3-none-any.whl → 0.2.12__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 (5) hide show
  1. titans_pytorch/neural_memory.py +1 -1
  2. {titans_pytorch-0.2.11.dist-info → titans_pytorch-0.2.12.dist-info}/METADATA +1 -1
  3. {titans_pytorch-0.2.11.dist-info → titans_pytorch-0.2.12.dist-info}/RECORD +5 -5
  4. {titans_pytorch-0.2.11.dist-info → titans_pytorch-0.2.12.dist-info}/WHEEL +0 -0
  5. {titans_pytorch-0.2.11.dist-info → titans_pytorch-0.2.12.dist-info}/licenses/LICENSE +0 -0
titans_pytorch/neural_memory.py CHANGED
@@ -288,7 +288,7 @@ class NeuralMemory(Module):
288
288
  attn_pool_chunks = False,
289
289
  momentum = True,
290
290
  pre_rmsnorm = True,
291
- post_rmsnorm = True,
291
+ post_rmsnorm = False,
292
292
  qk_rmsnorm = False,
293
293
  max_grad_norm: float | None = None,
294
294
  use_accelerated_scan = False,
{titans_pytorch-0.2.11.dist-info → titans_pytorch-0.2.12.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: titans-pytorch
3
- Version: 0.2.11
3
+ Version: 0.2.12
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.2.11.dist-info → titans_pytorch-0.2.12.dist-info}/RECORD RENAMED
@@ -2,8 +2,8 @@ titans_pytorch/__init__.py,sha256=Y3m_ZlpEqYwp-Md1ARhNGJxq8bQp8ty1o039nZOOJo0,27
2
2
  titans_pytorch/associative_scan.py,sha256=Y-iYqmFuG-NoCKu6kgql1mhowXTeJfyawi3eUIXamp0,2650
3
3
  titans_pytorch/mac_transformer.py,sha256=RfJ1SvQH5_4PmlB7g-13wPAqYtCCUJxfmtaL0oBrRCU,24563
4
4
  titans_pytorch/memory_models.py,sha256=Q9SAIyAbStF5Tz0EhvRbn3yAdE3nk3xKc1ndieIe714,4671
5
- titans_pytorch/neural_memory.py,sha256=1wX8dbGENHWk7sfz7IFF1G8KY4U5tsNh3cqSDxTUf2U,26150
6
- titans_pytorch-0.2.11.dist-info/METADATA,sha256=CMLW5FSamLp0cPhIohOD_yXjCXoxqCPzwJrA0e83vQE,6812
7
- titans_pytorch-0.2.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
8
- titans_pytorch-0.2.11.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
9
- titans_pytorch-0.2.11.dist-info/RECORD,,
5
+ titans_pytorch/neural_memory.py,sha256=PbPjpm7BZIkl1z4Zyt3OcWW6MxWKAnTlmDvJthcmlp4,26151
6
+ titans_pytorch-0.2.12.dist-info/METADATA,sha256=xZvhE0thLfasfF7DHO2FcwNo-bD0T5_R01pvL4sWd4A,6812
7
+ titans_pytorch-0.2.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
8
+ titans_pytorch-0.2.12.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
9
+ titans_pytorch-0.2.12.dist-info/RECORD,,