poipiii commited on
Commit
10ada40
·
1 Parent(s): 1b511a3

test latent fix

Browse files
Files changed (1) hide show
  1. 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)