diff --git a/Train-Runpod.ipynb b/Train-Runpod.ipynb
index 48371e1..497e3b4 100644
--- a/Train-Runpod.ipynb
+++ b/Train-Runpod.ipynb
@@ -73,6 +73,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "!pip install --upgrade jupyterlab\n",
+    "!pip install --upgrade ipywidgets\n",
     "!pip install -q omegaconf\n",
     "!pip install -q einops\n",
     "!pip install -q pytorch-lightning==1.6.5\n",
@@ -86,7 +88,6 @@
     "!pip install -q torchmetrics==0.6.0\n",
     "!pip install -e .\n",
     "!pip install huggingface_hub\n",
-    "!pip install ipywidgets==7.7.1\n",
     "import ipywidgets as widgets\n",
     "print(finish_msg)"
    ]
@@ -440,7 +441,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.13"
+   "version": "3.8.13 (default, Oct 19 2022, 22:38:03) [MSC v.1916 64 bit (AMD64)]"
   },
   "vscode": {
    "interpreter": {