Spaces:
Runtime error
Runtime error
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}")
|