x-transformers 1.44.4__tar.gz → 1.44.5__tar.gz

{x_transformers-1.44.4/x_transformers.egg-info → x_transformers-1.44.5}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: x-transformers
-Version: 1.44.4
+Version: 1.44.5
 Summary: X-Transformers - Pytorch
 Home-page: https://github.com/lucidrains/x-transformers
 Author: Phil Wang
@@ -19,3 +19,12 @@ Requires-Dist: einops>=0.8.0
 Requires-Dist: loguru
 Requires-Dist: packaging>=21.0
 Requires-Dist: torch>=2.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: summary

{x_transformers-1.44.4 → x_transformers-1.44.5}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 setup(
   name = 'x-transformers',
   packages = find_packages(exclude=['examples']),
-  version = '1.44.4',
+  version = '1.44.5',
   license='MIT',
   description = 'X-Transformers - Pytorch',
   author = 'Phil Wang',

{x_transformers-1.44.4 → x_transformers-1.44.5}/x_transformers/x_transformers.py

@@ -2224,7 +2224,7 @@ class AttentionLayers(Module):
         layer_variables = tuple(tuple(layer_variable[i] for i in layers_execute_order) for layer_variable in layer_variables)
 
         # derived input for reinjection if needed
-
+        inp_inject = None
         if self.reinject_input:
             assert not exists(in_attn_cond)
             inp_inject = self.reinject_input_proj(x)
@@ -2282,7 +2282,7 @@ class AttentionLayers(Module):
                 post_branch_norm = maybe(partial)(post_branch_norm, **norm_kwargs)
                 post_main_norm = maybe(partial)(post_main_norm, **norm_kwargs)
 
-            if self.reinject_input:
+            if exists(inp_inject):
                 x = x + inp_inject
 
             if exists(pre_norm):

{x_transformers-1.44.4 → x_transformers-1.44.5/x_transformers.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: x-transformers
-Version: 1.44.4
+Version: 1.44.5
 Summary: X-Transformers - Pytorch
 Home-page: https://github.com/lucidrains/x-transformers
 Author: Phil Wang
@@ -19,3 +19,12 @@ Requires-Dist: einops>=0.8.0
 Requires-Dist: loguru
 Requires-Dist: packaging>=21.0
 Requires-Dist: torch>=2.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: summary