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