poipiii commited on
Commit
5c7f092
1 Parent(s): fe203fe

test in latnent upcale

Browse files
Files changed (1) hide show
  1. pipeline.py +8 -0
pipeline.py CHANGED
@@ -841,6 +841,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
841
  callback(i, t, latents)
842
  if is_cancelled_callback is not None and is_cancelled_callback():
843
  return None
 
844
  print(latents)
845
  print(latents.shape)
846
  latents = torch.nn.functional.interpolate(
@@ -850,6 +851,13 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
850
  # expand the latents if we are doing classifier free guidance
851
  latent_model_input = torch.cat(
852
  [latents] * 2) if do_classifier_free_guidance else latents
 
 
 
 
 
 
 
853
  latent_model_input = self.scheduler.scale_model_input(
854
  latent_model_input, t)
855
 
 
841
  callback(i, t, latents)
842
  if is_cancelled_callback is not None and is_cancelled_callback():
843
  return None
844
+ print("after first step denoise latents")
845
  print(latents)
846
  print(latents.shape)
847
  latents = torch.nn.functional.interpolate(
 
851
  # expand the latents if we are doing classifier free guidance
852
  latent_model_input = torch.cat(
853
  [latents] * 2) if do_classifier_free_guidance else latents
854
+ print("latent_model_input")
855
+ print(latent_model_input)
856
+ print(latent_model_input.shape)
857
+
858
+ print("2nd step timestep")
859
+ print(t)
860
+
861
  latent_model_input = self.scheduler.scale_model_input(
862
  latent_model_input, t)
863