titans-pytorch 0.0.17__tar.gz → 0.0.18__tar.gz

{titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: titans-pytorch
-Version: 0.0.17
+Version: 0.0.18
 Summary: Titans
 Project-URL: Homepage, https://pypi.org/project/titans-pytorch/
 Project-URL: Repository, https://github.com/lucidrains/titans-pytorch

{titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "titans-pytorch"
-version = "0.0.17"
+version = "0.0.18"
 description = "Titans"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }

{titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/titans_pytorch/titans.py

@@ -269,7 +269,7 @@ class NeuralMemory(Module):
                 gates = F.pad(gates, (0, next_power_two_seq_len - seq_len))
                 inputs = F.pad(inputs, (0, next_power_two_seq_len - seq_len))
 
-                outputs = scan(gates, inputs)
+                outputs = scan(gates.contiguous(), inputs.contiguous())
 
                 outputs = outputs[..., :seq_len]
                 outputs = rearrange(outputs, 'b d n -> b n d')

{titans_pytorch-0.0.17 → titans_pytorch-0.0.18}/train.py

@@ -15,7 +15,6 @@ from taylor_series_linear_attention import TaylorSeriesLinearAttn
 
 from titans_pytorch.titans import (
     NeuralMemory,
-    MemoryAttention,
     MemoryMLP
 )