poipiii
commited on
Commit
·
7ab41fe
1
Parent(s):
662dbef
test in latnent upcale
Browse files- 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:
|