safe log

lucidrains · Dec 2, 2024 · 48afc51 · 48afc51
1 parent 35452f4
commit 48afc51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'x-transformers',
   packages = find_packages(exclude=['examples']),
-  version = '1.42.21',
+  version = '1.42.22',
   license='MIT',
   description = 'X-Transformers - Pytorch',
   author = 'Phil Wang',

diff --git a/x_transformers/x_transformers.py b/x_transformers/x_transformers.py
@@ -1465,7 +1465,7 @@ def forward(
         # laser
 
         if self.laser:
-            out = out.log() + values_max
+            out = log(out) + values_max
 
         # store the values for resformer or Neutreno