Chaerin5 commited on
Commit
b4246cf
·
1 Parent(s): 5e1df3e

enable zerogpu

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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=120)
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