daviddoji/pbdl-book
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improved pip install for probmodels-ddpm-fm

Browse Source
This commit is contained in:
N_T 2024-12-09 14:24:48 +08:00
parent fe1393fcd1
commit 148118cbe3
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
probmodels-ddpm-fm.ipynb
View File

@ -76,7 +76,12 @@
} }
], ],
"source": [ "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", "!git clone https://github.com/tum-pbs/Diffusion-based-Flow-Prediction.git\n",
"%cd Diffusion-based-Flow-Prediction/" "%cd Diffusion-based-Flow-Prediction/"
] ]