Fixed bug in PR sokrypton#633

ntnn19 · Jul 25, 2024 · c0d371d · c0d371d
1 parent 3f85170
commit c0d371d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/colabfold/batch.py b/colabfold/batch.py
@@ -1,3 +1,4 @@
+
 from __future__ import annotations
 
 import os
@@ -373,6 +374,9 @@ def predict_structure(
                     # TODO: add pad_input_mulitmer()
                     input_features = feature_dict
                     input_features["asym_id"] = input_features["asym_id"] - input_features["asym_id"][...,0]
+                    input_features['chain_num'] = input_features["asym_id"].max() + 1
+                    model_runner.chain_num = input_features['chain_num']
+
             else:
                 if model_num == 0:
                     input_features = model_runner.process_features(feature_dict, random_seed=seed)