From c1653f5efc6a2d0f64c85d949efb76d7d29ff3b2 Mon Sep 17 00:00:00 2001
From: exponentialXP <127988888+exponentialXP@users.noreply.github.com>
Date: Sat, 17 Feb 2024 15:23:42 +0000
Subject: [PATCH] Update Diffusion Model.ipynb

---
 01_Diffusion_Models_Tutorial/Diffusion Model.ipynb | 1 +
 1 file changed, 1 insertion(+)

diff --git a/01_Diffusion_Models_Tutorial/Diffusion Model.ipynb b/01_Diffusion_Models_Tutorial/Diffusion Model.ipynb
index 8078f38..4c32132 100644
--- a/01_Diffusion_Models_Tutorial/Diffusion Model.ipynb	
+++ b/01_Diffusion_Models_Tutorial/Diffusion Model.ipynb	
@@ -907,6 +907,7 @@
     "        t = torch.full((1,), i, dtype=torch.long, device=device)\n",
     "        labels = torch.tensor([c] * NUM_DISPLAY_IMAGES).resize(NUM_DISPLAY_IMAGES, 1).float().to(device)\n",
     "        imgs = diffusion_model.backward(x=imgs, t=t, model=unet.eval().to(device), labels = labels)\n",
+    "    imgs = imgs.clamp(-1, 1)\n",
     "    for idx, img in enumerate(imgs):\n",
     "        ax[c][idx].imshow(reverse_transform(img))\n",
     "        ax[c][idx].set_title(f\"Class: {classes[c]}\", fontsize = 100)\n",