oops

Author: lucidrains

Diff for: alphafold3_pytorch/attention.py

@@ -256,7 +256,7 @@ def forward(
         orig_v = v
 
         if exists(value_residual):
-            v = v + value_residual
+            v = 0.5 * (v + value_residual)
 
         # split heads