Skip to content

Commit 769fd14

Browse files
committed
follow the alphafold convention of inner attention gating
1 parent 39cf9fc commit 769fd14

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

alphafold3_pytorch/alphafold3.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1355,6 +1355,7 @@ def __init__(
13551355
linear_attn = TaylorSeriesLinearAttn(
13561356
dim = dim,
13571357
prenorm = True,
1358+
gate_value_heads = True,
13581359
**linear_attn_kwargs
13591360
)
13601361

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[project]
22
name = "alphafold3-pytorch"
3-
version = "0.0.52"
3+
version = "0.0.53"
44
description = "Alphafold 3 - Pytorch"
55
authors = [
66
{ name = "Phil Wang", email = "[email protected]" }

0 commit comments

Comments
 (0)