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

test in latnent upcale

Browse files
Files changed (1) hide show
  1. pipeline.py +6 -2
pipeline.py CHANGED
@@ -860,11 +860,15 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
860
 
861
  latent_model_input = self.scheduler.scale_model_input(
862
  latent_model_input, t)
863
-
 
 
864
  # predict the noise residual
865
  noise_pred = self.unet(latent_model_input, t,
866
  encoder_hidden_states=text_embeddings).sample
867
-
 
 
868
  # perform guidance
869
  if do_classifier_free_guidance:
870
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
 
860
 
861
  latent_model_input = self.scheduler.scale_model_input(
862
  latent_model_input, t)
863
+ print("latent_model_input after scheduler")
864
+ print(latent_model_input)
865
+ print(latent_model_input.shape)
866
  # predict the noise residual
867
  noise_pred = self.unet(latent_model_input, t,
868
  encoder_hidden_states=text_embeddings).sample
869
+ print("noise_pred")
870
+ print(noise_pred)
871
+ print(noise_pred.shape)
872
  # perform guidance
873
  if do_classifier_free_guidance:
874
  noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)