Fix inference script again

NatnaelT · Jun 23, 2022 · 3654b7f · 3654b7f
1 parent c22d011
commit 3654b7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_pretrained_openfold.py b/run_pretrained_openfold.py
@@ -342,7 +342,7 @@ def main(args):
         )
 
         for model, model_version in load_models_from_command_line(args, config):
-            working_batch = deepcopy(batch)
+            working_batch = deepcopy(processed_feature_dict)
             out = run_model(model, working_batch, tag, args)
 
             # Toss out the recycling dimensions --- we don't need them anymore