Properly reading the embedding file

5a8d2b78 · sachinkadyan7 · Sachin Kadyan · 36d5708c · 5a8d2b78
Commit 5a8d2b78 authored Sep 11, 2023 by sachinkadyan7 Committed by Sachin Kadyan Oct 10, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openfold/data/data_pipeline.py openfold/data/data_pipeline.py +1 -1

No files found.
--- a/openfold/data/data_pipeline.py
+++ b/openfold/data/data_pipeline.py
@@ -652,7 +652,7 @@ class DataPipeline:
            if (ext == ".pt"):
                # Load embedding file
                seqemb_data = torch.load(path)
-                seqemb_features["seq_embedding"] = seqemb_data
+                seqemb_features["seq_embedding"] = seqemb_data["representations"][33]

        return seqemb_features