titans-pytorch 0.0.24__tar.gz → 0.0.25__tar.gz

{titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/PKG-INFO

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

{titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/pyproject.toml

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

{titans_pytorch-0.0.24 → titans_pytorch-0.0.25}/titans_pytorch/titans.py

@@ -136,6 +136,7 @@ class NeuralMemory(Module):
         )
     ):
         super().__init__()
+        dim_head = default(dim_head, dim)
 
         # norms
 
@@ -146,7 +147,6 @@ class NeuralMemory(Module):
 
         # maybe multi-headed
 
-        dim_head = default(dim_head, dim)
         dim_inner = dim_head * heads
 
         self.heads = heads