cocktailpeanut commited on
Commit
c12474d
1 Parent(s): 84677d0
Files changed (1) hide show
  1. pipelines/sdxl_instantir.py +1 -0
pipelines/sdxl_instantir.py CHANGED
@@ -931,6 +931,7 @@ class InstantIRPipeline(
931
 
932
  @torch.no_grad()
933
  def init_latents(self, latents, generator, timestep):
 
934
  noise = torch.randn(latents.shape, generator=generator, device=self.vae.device, dtype=self.vae.dtype, layout=torch.strided)
935
  bsz = latents.shape[0]
936
  print(f"init latent at {timestep}")
 
931
 
932
  @torch.no_grad()
933
  def init_latents(self, latents, generator, timestep):
934
+ print(f"device={self.vae.device}, generator={generator}")
935
  noise = torch.randn(latents.shape, generator=generator, device=self.vae.device, dtype=self.vae.dtype, layout=torch.strided)
936
  bsz = latents.shape[0]
937
  print(f"init latent at {timestep}")