Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1215,6 +1215,8 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
 
                 if use_KDS:
 
+                    print("using KDS")
+
                     # 2) Compute x₀ prediction
                     beta_t   = 1 - self.scheduler.alphas_cumprod[t]
                     alpha_t  = self.scheduler.alphas_cumprod[t].sqrt()
@@ -1244,6 +1246,9 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                       + sigma_prev * noise_pred_kds
                     ).detach().requires_grad_(True)
                 else:
+
+                    print("KDS Free")
+                    
                     # compute the previous noisy sample x_t -> x_t-1
                     latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs, return_dict=False)[0]