Added flag in training script for using sequence embeddings

train_openfold.py

@@ -416,6 +416,10 @@ if __name__ == "__main__":
         help='''Cutoff for all templates. In training mode, templates are also 
                 filtered by the release date of the target'''
     )
+    parser.add_argument(
+        "--use_single_seq_mode", type=str, default=None,
+        help="Use single sequence embeddings instead of MSAs."
+    )
     parser.add_argument(
         "--distillation_data_dir", type=str, default=None,
         help="Directory containing training PDB files"