Spaces:
Running
on
Zero
Running
on
Zero
enable zerogpu
Browse files
app.py
CHANGED
@@ -322,11 +322,12 @@ def get_ref_anno(ref):
|
|
322 |
hand_mask = np.zeros_like(img[:,:, 0])
|
323 |
ref_pose = np.zeros_like(img)
|
324 |
|
325 |
-
@spaces.GPU(duration=
|
326 |
def make_ref_cond(
|
327 |
image
|
328 |
):
|
329 |
print("ready to run autoencoder")
|
|
|
330 |
latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
|
331 |
return image[None, ...], latent
|
332 |
|
|
|
322 |
hand_mask = np.zeros_like(img[:,:, 0])
|
323 |
ref_pose = np.zeros_like(img)
|
324 |
|
325 |
+
@spaces.GPU(duration=60)
|
326 |
def make_ref_cond(
|
327 |
image
|
328 |
):
|
329 |
print("ready to run autoencoder")
|
330 |
+
image = image.cuda()
|
331 |
latent = opts.latent_scaling_factor * autoencoder.encode(image[None, ...]).sample()
|
332 |
return image[None, ...], latent
|
333 |
|