diff --git a/nbs/finetune.ipynb b/nbs/finetune.ipynb
index 8e8cfc7..70208f1 100644
--- a/nbs/finetune.ipynb
+++ b/nbs/finetune.ipynb
@@ -49,10 +49,10 @@
         ")\n",
         "\n",
         "# @markdown To get your API key visit https://dreamstudio.ai/account\n",
-        "STABILITY_HOST = \"grpc-staging.stability.ai:443\" #@param {type:\"string\"}\n",
+        "STABILITY_HOST = \"grpc.stability.ai:443\"  #@param [\"grpc.stability.ai:443\", \"grpc-staging.stability.ai:443\"] {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\", \"stable-diffusion-xl-1024-v1-0\"] {type:\"string\"}\n",
+        "engine_id = \"stable-diffusion-xl-1024-v1-0\" #@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",
@@ -277,7 +277,7 @@
       "name": "python",
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
-      "version": "3.11.4"
+      "version": "3.11.2"
     },
     "orig_nbformat": 4
   },