improved pip install for probmodels-ddpm-fm

2024-12-09 14:24:48 +08:00 · 2024-12-09 14:24:48 +08:00 · 148118cbe3
commit 148118cbe3
parent fe1393fcd1
1 changed files with 6 additions and 1 deletions
--- 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/"
   ]