Update alphafold3.py

lucidrains · Oct 4, 2024 · 26c1ec4 · 26c1ec4
1 parent b6c8b17
commit 26c1ec4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alphafold3_pytorch/alphafold3.py b/alphafold3_pytorch/alphafold3.py
@@ -529,7 +529,7 @@ def __init__(
         self,
         *,
         dim,
-        expansion_factor = 4
+        expansion_factor = 2
     ):
         super().__init__()
         dim_inner = int(dim * expansion_factor)