Fix bug in evoformer

484ce867 · Gustaf Ahdritz · GitHub · 5eb86a78 · 484ce867
Unverified Commit 484ce867 authored Feb 17, 2022 by Gustaf Ahdritz Committed by GitHub Feb 17, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openfold/model/evoformer.py openfold/model/evoformer.py +1 -1

No files found.
--- a/openfold/model/evoformer.py
+++ b/openfold/model/evoformer.py
@@ -571,7 +571,7 @@ class ExtraMSAStack(nn.Module):
                pair_dropout=pair_dropout,
                inf=inf,
                eps=eps,
-                ckpt=False,
+                ckpt=ckpt,
            )
            self.blocks.append(block)