Spaces:
Running
Running
cocktailpeanut
commited on
Commit
•
c12474d
1
Parent(s):
84677d0
update
Browse files
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}")
|