diff --git a/nbs/finetune.ipynb b/nbs/finetune.ipynb
index 00bab28..8e8cfc7 100644
--- a/nbs/finetune.ipynb
+++ b/nbs/finetune.ipynb
@@ -2,7 +2,7 @@
   "cells": [
     {
       "cell_type": "code",
-      "execution_count": 24,
+      "execution_count": 1,
       "metadata": {
         "cellView": "form",
         "id": "T9ma2X7bhH8y"
@@ -52,7 +52,7 @@
         "STABILITY_HOST = \"grpc-staging.stability.ai:443\" #@param {type:\"string\"}\n",
         "STABILITY_KEY = getpass.getpass('Enter your API Key')\n",
         "\n",
-        "engine_id = \"stable-diffusion-xl-1024-v0-9\" #@param [\"stable-diffusion-xl-1024-v0-9\"] {type:\"string\"}\n",
+        "engine_id = \"stable-diffusion-xl-1024-v0-9\" #@param [\"stable-diffusion-xl-1024-v0-9\", \"stable-diffusion-xl-1024-v1-0\"] {type:\"string\"}\n",
         "\n",
         "# Create API context to query user info and generate images\n",
         "context = Context(STABILITY_HOST, STABILITY_KEY, generate_engine_id=engine_id)\n",
@@ -256,13 +256,6 @@
         "        if artifact.type == generation.ARTIFACT_IMAGE:\n",
         "            display(Image.open(io.BytesIO(artifact.binary)))"
       ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {},
-      "outputs": [],
-      "source": []
     }
   ],
   "metadata": {
@@ -284,7 +277,7 @@
       "name": "python",
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
-      "version": "3.9.7"
+      "version": "3.11.4"
     },
     "orig_nbformat": 4
   },