Spaces:
Runtime error
Runtime error
enable zerogpu
Browse files
app.py
CHANGED
@@ -262,7 +262,7 @@ def make_ref_cond(
|
|
262 |
print("ready to run autoencoder")
|
263 |
# print(f"image.device: {image.device}, type(image): {type(image)}")
|
264 |
# image = image.to("cuda")
|
265 |
-
print(f"autoencoder
|
266 |
latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
|
267 |
return image[None, ...], latent
|
268 |
|
|
|
262 |
print("ready to run autoencoder")
|
263 |
# print(f"image.device: {image.device}, type(image): {type(image)}")
|
264 |
# image = image.to("cuda")
|
265 |
+
print(f"autoencoder device: {next(autoencoder.parameters()).device}")
|
266 |
latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
|
267 |
return image[None, ...], latent
|
268 |
|