fix quaternion update

lucidrains · Aug 9, 2021 · c39e31e · c39e31e
1 parent b88e9e3
commit c39e31e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/alphafold2_pytorch/alphafold2.py b/alphafold2_pytorch/alphafold2.py
@@ -859,7 +859,7 @@ def forward(
                 quaternion_update, translation_update = self.to_quaternion_update(single_repr).chunk(2, dim = -1)
                 quaternion_update = F.pad(quaternion_update, (1, 0), value = 1.)
 
-                quaternions = quaternions + quaternion_multiply(quaternions, quaternion_update)
+                quaternions = quaternion_multiply(quaternions, quaternion_update)
                 translations = translations + einsum('b n c, b n c r -> b n r', translation_update, rotations)
 
             points_local = self.to_points(single_repr)

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'alphafold2-pytorch',
   packages = find_packages(),
-  version = '0.4.24',
+  version = '0.4.25',
   license='MIT',
   description = 'AlphaFold2 - Pytorch',
   author = 'Phil Wang, Eric Alcaide',