diff --git a/probmodels-ddpm-fm.ipynb b/probmodels-ddpm-fm.ipynb
index a4dd102..b96a25a 100644
--- a/probmodels-ddpm-fm.ipynb
+++ b/probmodels-ddpm-fm.ipynb
@@ -76,7 +76,12 @@
     }
    ],
    "source": [
-    "%pip install --upgrade --quiet einops bayesian_torch\n",
+    "try:\n",
+    "    import google.colab  # to ensure that we are inside colab\n",
+    "    !pip install --upgrade --quiet einops bayesian_torch\n",
+    "except ImportError:\n",
+    "    print('This notebook is running locally, please make sure the necessary pip packages are installed.')\n",
+    "    pass\n",
     "!git clone https://github.com/tum-pbs/Diffusion-based-Flow-Prediction.git\n",
     "%cd Diffusion-based-Flow-Prediction/"
    ]