poipiii
commited on
Commit
·
10ada40
1
Parent(s):
1b511a3
test latent fix
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
@@ -633,6 +633,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
|
|
633 |
else:
|
634 |
latents = torch.randn(shape, generator=generator, device=device, dtype=dtype)
|
635 |
else:
|
|
|
636 |
if latents.shape != shape:
|
637 |
raise ValueError(f"Unexpected latents shape, got {latents.shape}, expected {shape}")
|
638 |
latents = latents.to(device)
|
|
|
633 |
else:
|
634 |
latents = torch.randn(shape, generator=generator, device=device, dtype=dtype)
|
635 |
else:
|
636 |
+
print(latents)
|
637 |
if latents.shape != shape:
|
638 |
raise ValueError(f"Unexpected latents shape, got {latents.shape}, expected {shape}")
|
639 |
latents = latents.to(device)
|