diff --git a/cellfinder/__init__.py b/cellfinder/__init__.py
index 6df547b6..fee55e19 100644
--- a/cellfinder/__init__.py
+++ b/cellfinder/__init__.py
@@ -26,6 +26,7 @@
 
 # Set the Keras backend to torch
 os.environ["KERAS_BACKEND"] = "torch"
+os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 
 __license__ = "BSD-3-Clause"