From 76ea9c856f99f4c8eca45a0b1801593bb982584b Mon Sep 17 00:00:00 2001
From: Clarence Chen <clarencechenct@gmail.com>
Date: Sun, 12 Mar 2023 15:04:00 -0700
Subject: [PATCH] Log loss for debugging

---
 .../stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py
index feb5bbcf9d..c2132a46c5 100644
--- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py
+++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py
@@ -1028,6 +1028,7 @@ class StableDiffusionPix2PixZeroPipeline(DiffusionPipeline):
 
                 # call the callback, if provided
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
+                    progress_bar.set_postfix({"loss": loss.loss.detach().item()})
                     progress_bar.update()
 
         # 11. Post-process the latents.