Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "56277ea05634009765b8bd296c470776fd92149c"
Commit c4d9f57f authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Reintroduce in-place operation to attention module

parent 6c5cd945
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
openfold/model/primitives.py
View file @ c4d9f57f
...@@ -269,7 +269,7 @@ class Attention(nn.Module): ...@@ -269,7 +269,7 @@ class Attention(nn.Module):
if biases is not None: if biases is not None:
for b in biases: for b in biases:
a = a + b a += b
a = self.softmax(a) a = self.softmax(a)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment