Update Stable_Diffusion2.ipynb

Author: cedro3

Stable_Diffusion2.ipynb

@@ -122,10 +122,10 @@
         "        init_image = init_image.resize((512, 512))\n",
         "        init_image = preprocess(init_image)\n",
         "        with autocast(\"cuda\"):\n",
-        "            images = pipeimg([prompt] * num_samples, init_image=init_image, strength=strength, guidance_scale=7.5)[\"sample\"]\n",
+        "            images = pipeimg([prompt] * num_samples, init_image=init_image, strength=strength, guidance_scale=7.5)[0]\n",
         "    else: \n",
         "        with autocast(\"cuda\"):\n",
-        "            images = pipe([prompt] * num_samples, guidance_scale=7.5)[\"sample\"]\n",
+        "            images = pipe([prompt] * num_samples, guidance_scale=7.5)[0]\n",
         "\n",
         "    return images\n",
         "\n",
@@ -199,4 +199,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}