titans-pytorch 0.2.8__py3-none-any.whl → 0.2.9__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 +2 -3
  2. {titans_pytorch-0.2.8.dist-info → titans_pytorch-0.2.9.dist-info}/METADATA +1 -1
  3. {titans_pytorch-0.2.8.dist-info → titans_pytorch-0.2.9.dist-info}/RECORD +5 -5
  4. {titans_pytorch-0.2.8.dist-info → titans_pytorch-0.2.9.dist-info}/WHEEL +0 -0
  5. {titans_pytorch-0.2.8.dist-info → titans_pytorch-0.2.9.dist-info}/licenses/LICENSE +0 -0
titans_pytorch/neural_memory.py CHANGED
@@ -742,7 +742,7 @@ class NeuralMemory(Module):
742
742
 
743
743
  # retrieve
744
744
 
745
- retrieved = self.retrieve_memories(token, weights, chunk_size = 1)
745
+ retrieved = self.retrieve_memories(token, updates, chunk_size = 1)
746
746
 
747
747
  # next state tuple
748
748
 
@@ -801,10 +801,9 @@ class NeuralMemory(Module):
801
801
 
802
802
  updates = updates.apply(lambda t: pad_at_dim(t, (1, 0), dim = 1))
803
803
 
804
-
805
804
  retrieved = self.retrieve_memories(
806
805
  seq,
807
- mem_model_weights,
806
+ updates,
808
807
  chunk_size = chunk_size,
809
808
  prev_layer_updates = prev_layer_updates
810
809
  )
{titans_pytorch-0.2.8.dist-info → titans_pytorch-0.2.9.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: titans-pytorch
3
- Version: 0.2.8
3
+ Version: 0.2.9
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.8.dist-info → titans_pytorch-0.2.9.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=UOJAMv7nTgkefBB7M7K3U0NnFkz75tFRG5WLXRdfnLw,26039
4
4
  titans_pytorch/memory_models.py,sha256=Q9SAIyAbStF5Tz0EhvRbn3yAdE3nk3xKc1ndieIe714,4671
5
- titans_pytorch/neural_memory.py,sha256=WAeR-nOpy1XbBP590By1-tCgirulqPbFGut4H1B77-g,24910
6
- titans_pytorch-0.2.8.dist-info/METADATA,sha256=4fLUv34KqloeYMWjHBUmp-3iEw0Xq47fjRrwlkyTEsM,6811
7
- titans_pytorch-0.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
8
- titans_pytorch-0.2.8.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
9
- titans_pytorch-0.2.8.dist-info/RECORD,,
5
+ titans_pytorch/neural_memory.py,sha256=YVbKl7DYKFWUgCawDTxXIEgJAcl7nq5OaZytmovIl8Q,24899
6
+ titans_pytorch-0.2.9.dist-info/METADATA,sha256=fSFt54zXLKB5gRhLTJd9551O0pF2qcYNlR7039yJiD0,6811
7
+ titans_pytorch-0.2.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
8
+ titans_pytorch-0.2.9.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
9
+ titans_pytorch-0.2.9.dist-info/RECORD,,