Fix bug in import script

cf041c7e · Gustaf Ahdritz · 91f031f5 · cf041c7e
Commit cf041c7e authored Jul 03, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openfold/utils/import_weights.py openfold/utils/import_weights.py +1 -1

No files found.
--- a/openfold/utils/import_weights.py
+++ b/openfold/utils/import_weights.py
@@ -393,7 +393,7 @@ def generate_translation_dict(model, version):
        "model_5_ptm",
    ]

-    if version in no_templ:
+    if version not in no_templ:
        tps_blocks = model.template_pair_stack.blocks
        tps_blocks_params = stacked(
            [TemplatePairBlockParams(b) for b in tps_blocks]