Fix typo in model.py

e1142cf3 · Gustaf Ahdritz · GitHub · ca43b8dd · e1142cf3
Unverified Commit e1142cf3 authored Dec 28, 2021 by Gustaf Ahdritz Committed by GitHub Dec 28, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/openfold/model/model.py
+++ b/openfold/model/model.py
@@ -316,7 +316,7 @@ class AlphaFold(nn.Module):
        # [*, N, C_m]
        m_1_prev = m[..., 0, :, :]
-        # [* N, N, C_z]
+        # [*, N, N, C_z]
        z_prev = z
        # [*, N, 3]