Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
8e8455b
1
Parent(s):
7d670d5
hf cuda issue
Browse files
app.py
CHANGED
@@ -852,9 +852,11 @@ def sample_inpaint(
|
|
852 |
jump_length=jump_length,
|
853 |
jump_n_sample=jump_n_sample,
|
854 |
).chunk(2)
|
|
|
855 |
sampled_images = autoencoder.decode(samples / opts.latent_scaling_factor)
|
856 |
sampled_images = torch.clamp(sampled_images, min=-1.0, max=1.0)
|
857 |
sampled_images = unnormalize(sampled_images.permute(0, 2, 3, 1).cpu().numpy())
|
|
|
858 |
|
859 |
# visualize
|
860 |
results = []
|
|
|
852 |
jump_length=jump_length,
|
853 |
jump_n_sample=jump_n_sample,
|
854 |
).chunk(2)
|
855 |
+
print("inpaint loop finished")
|
856 |
sampled_images = autoencoder.decode(samples / opts.latent_scaling_factor)
|
857 |
sampled_images = torch.clamp(sampled_images, min=-1.0, max=1.0)
|
858 |
sampled_images = unnormalize(sampled_images.permute(0, 2, 3, 1).cpu().numpy())
|
859 |
+
print("inpaint decode finished")
|
860 |
|
861 |
# visualize
|
862 |
results = []
|