From ecb8be97aa688961d56cf4682848751e6aded19c Mon Sep 17 00:00:00 2001
From: Victor Hall <victor.charles.hall@gmail.com>
Date: Wed, 28 Dec 2022 13:01:51 -0500
Subject: [PATCH] fix runpod

---
 Train-Runpod.ipynb | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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": {