Spaces:
Runtime error
Runtime error
enable zerogpu
Browse files
app.py
CHANGED
|
@@ -256,7 +256,6 @@ hands = mp_hands.Hands(
|
|
| 256 |
min_detection_confidence=0.1,
|
| 257 |
)
|
| 258 |
|
| 259 |
-
@spaces.GPU
|
| 260 |
def make_ref_cond(
|
| 261 |
image
|
| 262 |
):
|
|
@@ -264,7 +263,6 @@ def make_ref_cond(
|
|
| 264 |
# print(f"image.device: {image.device}, type(image): {type(image)}")
|
| 265 |
# image = image.to("cuda")
|
| 266 |
print(f"autoencoder device: {next(autoencoder.parameters()).device}")
|
| 267 |
-
image = image.to(next(autoencoder.parameters()).device)
|
| 268 |
latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
|
| 269 |
return image[None, ...], latent
|
| 270 |
|
|
|
|
| 256 |
min_detection_confidence=0.1,
|
| 257 |
)
|
| 258 |
|
|
|
|
| 259 |
def make_ref_cond(
|
| 260 |
image
|
| 261 |
):
|
|
|
|
| 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 |
|