poipiii
commited on
Commit
•
0efbf9b
1
Parent(s):
5c7f092
test in latnent upcale
Browse files- 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)
|