chaerinmin commited on
Commit
8e8455b
·
1 Parent(s): 7d670d5

hf cuda issue

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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 = []