cpu support only

utils_app.py

@@ -112,7 +112,7 @@ def run_inference(params):
     )
 
     model = get_model(params['model'], params)
-    print(model.hparams)
+    # print(model.hparams)
     model.df_test_path = params['df_test_path']
     model.external_test = True
     model.hparams['use_fp'] = False
@@ -121,9 +121,9 @@ def run_inference(params):
     
     # Init trainer
     trainer = Trainer(
-        accelerator=params['accelerator'],
-        devices=params['devices'],
-        default_root_dir=params['experiment_dir']
+        accelerator='cpu',
+        devices=1,
+        default_root_dir=params['experiment_dir'],
     )
 
     # Prepare data module to test
@@ -142,5 +142,4 @@ if __name__ == "__main__":
     mass_diff_thresh = 20
     dataset_pth, subformula_dir = preprocess_spectra(mgf_path, model_choice, mass_diff_thresh=mass_diff_thresh)
     params = setup_config(model_choice, dataset_pth, candidates_pth, subformula_dir)
-    print(params)
     run_inference(params)