feat: Add explicit model selection in notebook

Stability-AI · Sep 6, 2022 · c43c6d9 · c43c6d9
1 parent 4e48732
commit c43c6d9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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",
         ")"
       ]