poipiii
commited on
Commit
•
144af9f
1
Parent(s):
5eed10a
test latent fix
Browse files- pipeline.py +5 -2
pipeline.py
CHANGED
@@ -809,6 +809,9 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
|
|
809 |
generator,
|
810 |
latents,
|
811 |
)
|
|
|
|
|
|
|
812 |
|
813 |
# 7. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
|
814 |
extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)
|
@@ -1032,8 +1035,8 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
|
|
1032 |
)
|
1033 |
|
1034 |
latents = torch.nn.functional.interpolate(latents, size=(int(height*resize_scale)//8, int(width*resize_scale)//8))
|
1035 |
-
print("noise")
|
1036 |
-
print(noise)
|
1037 |
|
1038 |
return self.__call__(
|
1039 |
prompt=prompt,
|
|
|
809 |
generator,
|
810 |
latents,
|
811 |
)
|
812 |
+
print("noise")
|
813 |
+
print(noise.shape)
|
814 |
+
print(noise)
|
815 |
|
816 |
# 7. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
|
817 |
extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)
|
|
|
1035 |
)
|
1036 |
|
1037 |
latents = torch.nn.functional.interpolate(latents, size=(int(height*resize_scale)//8, int(width*resize_scale)//8))
|
1038 |
+
# print("noise")
|
1039 |
+
# print(noise)
|
1040 |
|
1041 |
return self.__call__(
|
1042 |
prompt=prompt,
|