harsh99 commited on
Commit
915e59a
·
verified ·
1 Parent(s): 43be0ab

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -79,7 +79,7 @@ def prepare_inpainting_input(
79
  return noisy_latents
80
 
81
  # Compute VAE encodings
82
- def compute_vae_encodings(image_tensor, encoder, device="cuda"):
83
  """Encode image using VAE encoder"""
84
  # Generate random noise for encoding
85
  encoder_noise = torch.randn(
 
79
  return noisy_latents
80
 
81
  # Compute VAE encodings
82
+ def compute_vae_encodings(image_tensor, encoder, device="cpu"):
83
  """Encode image using VAE encoder"""
84
  # Generate random noise for encoding
85
  encoder_noise = torch.randn(