Merge pull request #8 from GT-LIT-Lab/device-fix

fall back to CPU if no GPUs are available

Author: TahaBinhuraib

encoding/models/nested_cv.py

@@ -87,6 +87,9 @@ def fit_predict(
                 device = "mps:0"
             elif torch.cuda.is_available():
                 device = "cuda"
+            else:
+                # if no GPU is available, fall back to CPU
+                device = "cpu"
         else:
              device = "cpu"
         logger.info(f"Using device: {device}")