poipiii commited on
Commit
7ab41fe
·
1 Parent(s): 662dbef

test in latnent upcale

Browse files
Files changed (1) hide show
  1. pipeline.py +5 -2
pipeline.py CHANGED
@@ -877,17 +877,20 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
877
  print("noise_pred")
878
  # print(noise_pred)
879
  print(noise_pred.shape)
 
 
880
  # perform guidance
881
  if do_classifier_free_guidance:
882
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
883
  noise_pred = noise_pred_uncond + guidance_scale * \
884
  (noise_pred_text - noise_pred_uncond)
885
-
 
886
  print("compute the previous noisy sample")
887
  # compute the previous noisy sample x_t -> x_t-1
888
  upscale_latents = self.scheduler.step(
889
  noise_pred, t, upscale_latents, **extra_step_kwargs).prev_sample
890
-
891
 
892
  print("compute mask")
893
  if mask is not None:
 
877
  print("noise_pred")
878
  # print(noise_pred)
879
  print(noise_pred.shape)
880
+
881
+ print("perform guidance")
882
  # perform guidance
883
  if do_classifier_free_guidance:
884
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
885
  noise_pred = noise_pred_uncond + guidance_scale * \
886
  (noise_pred_text - noise_pred_uncond)
887
+ print("noise_pred after guidance")
888
+ # print(noise_pred)
889
  print("compute the previous noisy sample")
890
  # compute the previous noisy sample x_t -> x_t-1
891
  upscale_latents = self.scheduler.step(
892
  noise_pred, t, upscale_latents, **extra_step_kwargs).prev_sample
893
+ print(upscale_latents.shape)
894
 
895
  print("compute mask")
896
  if mask is not None: