🐛 make CF Sklearn model run without MPS

- set fastai defaults to cpu
RasmussenLab · Feb 24, 2024 · bae51d7 · bae51d7
1 parent a4f36da
commit bae51d7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vaep/sklearn/cf_transformer.py b/vaep/sklearn/cf_transformer.py
@@ -81,7 +81,8 @@ def fit(self, X: pd.Series, y: pd.Series = None,
             y_range=(int(X.squeeze().min()),
                      int(X.squeeze().max()) + 1)
         )
-
+        if not cuda:
+            default_device(use=False)  # set to cpu
         if y is not None:
             X, frac = collab.combine_data(X, y)
         else: