Merge pull request fastai#21 from cly/fix-img2img

Update image2image example with correct start_step check
osamja · Nov 16, 2022 · 388f00c · 388f00c
2 parents cdb1c8f + eb78c69
commit 388f00c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Stable Diffusion Deep Dive.ipynb b/Stable Diffusion Deep Dive.ipynb
@@ -707,7 +707,7 @@
     "\n",
     "# Loop\n",
     "for i, t in tqdm(enumerate(scheduler.timesteps)):\n",
-    "    if i > start_step: # << This is the only modification to the loop we do\n",
+    "    if i >= start_step: # << This is the only modification to the loop we do\n",
     "        \n",
     "        # expand the latents if we are doing classifier-free guidance to avoid doing two forward passes.\n",
     "        latent_model_input = torch.cat([latents] * 2)\n",