Add workers arg to from_model_checkpoint (#1844)

Author: rayg1234

src/fairchem/core/calculate/ase_calculator.py

@@ -125,6 +125,7 @@ def from_model_checkpoint(
         overrides: dict | None = None,
         device: Literal["cuda", "cpu"] | None = None,
         seed: int = 41,
+        workers: int = 1,
     ) -> FAIRChemCalculator:
         """Instantiate a FAIRChemCalculator from a checkpoint file.
 
@@ -139,6 +140,7 @@ def from_model_checkpoint(
             overrides: Optional dictionary of settings to override default inference settings.
             device: Optional torch device to load the model onto.
             seed: Random seed for reproducibility.
+            workers: Number of parallel workers for prediction unit. Default is 1.
         """
 
         if name_or_path in pretrained_mlip.available_models:
@@ -147,13 +149,15 @@ def from_model_checkpoint(
                 inference_settings=inference_settings,
                 overrides=overrides,
                 device=device,
+                workers=workers,
             )
         elif os.path.isfile(name_or_path):
             predict_unit = pretrained_mlip.load_predict_unit(
                 name_or_path,
                 inference_settings=inference_settings,
                 overrides=overrides,
                 device=device,
+                workers=workers,
             )
         else:
             raise ValueError(