Spaces:
Runtime error
Runtime error
Update utils.py
Browse files
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="
|
| 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(
|