Merge branch 'main' of ssh://github.com/aqlaboratory/openfold into main

openfold/data/feature_pipeline.py

@@ -100,10 +100,8 @@ class FeaturePipeline:
     def __init__(
         self,
         config: ml_collections.ConfigDict,
-        params: Optional[Mapping[str, Mapping[str, np.ndarray]]] = None,
     ):
         self.config = config
-        self.params = params
 
     def process_features(
         self,

run_pretrained_openfold.py

@@ -48,7 +48,7 @@ def main(args):
     config = model_config(args.model_name)
     model = AlphaFold(config)
     model = model.eval()
-    import_jax_weights_(model, args.param_path)
+    import_jax_weights_(model, args.param_path, version=args.model_name)
     #script_preset_(model)
     model = model.to(args.model_device)