misnamed variable

lucidrains · Aug 3, 2021 · 23a739a · 23a739a
1 parent afe7ac9
commit 23a739a
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 = 'triangle-multiplicative-module',
   packages = find_packages(),
-  version = '0.0.2',
+  version = '0.0.3',
   license='MIT',
   description = 'Triangle Multiplicative Module',
   author = 'Phil Wang',

diff --git a/triangle_multiplicative_module/triangle_multiplicative_module.py b/triangle_multiplicative_module/triangle_multiplicative_module.py
@@ -67,7 +67,7 @@ def forward(self, x, mask = None):
         out_gate = self.out_gate(x).sigmoid()
 
         left = left * left_gate
-        right = right * left_gate
+        right = right * right_gate
 
         out = einsum(self.mix_einsum_eq, left, right)