From c43c6d9624cc1f0ba4bac0838cd487e1e25e272b Mon Sep 17 00:00:00 2001
From: Peter Salanki <peter@salanki.st>
Date: Fri, 26 Aug 2022 15:18:35 -0400
Subject: [PATCH] feat: Add explicit model selection in notebook

---
 nbs/demo_colab.ipynb | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/nbs/demo_colab.ipynb b/nbs/demo_colab.ipynb
index 60f5d60f..51f9d93e 100644
--- a/nbs/demo_colab.ipynb
+++ b/nbs/demo_colab.ipynb
@@ -62,7 +62,9 @@
         "\n",
         "\n",
         "stability_api = client.StabilityInference(\n",
-        "    key=os.environ['STABILITY_KEY'], \n",
+        "    key=os.environ['STABILITY_KEY'],\n",
+        "    # Optional: Specify a specific model/checkpoint. Will use the default model as defined in the Python SDK release currently used if not specified. \n",
+        "    engine=\"stable-diffusion-v1-5\",\n",
         "    verbose=True,\n",
         ")"
       ]