robinwitch commited on
Commit
88abf96
·
1 Parent(s): a30f1f1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -340,7 +340,7 @@ class BaseTrainer(object):
340
  return result
341
 
342
 
343
- @spaces.GPU(duration=80)
344
  def _warp(args,model, batch_data,joints,joint_mask_upper,joint_mask_hands,joint_mask_lower,use_trans,mean_upper,mean_hands,mean_lower,std_upper,std_hands,std_lower,trans_mean,trans_std):
345
  diffusion = create_gaussian_diffusion(use_ddim=args.use_ddim)
346
  args,model,vq_model_upper,vq_model_hands,vq_model_lower,mean_upper,mean_hands,mean_lower,std_upper,std_hands,std_lower,trans_mean,trans_std,vqvae_latent_scale=_warp_create_cuda_model(args,model,mean_upper,mean_hands,mean_lower,std_upper,std_hands,std_lower,trans_mean,trans_std)
 
340
  return result
341
 
342
 
343
+ @spaces.GPU(duration=60)
344
  def _warp(args,model, batch_data,joints,joint_mask_upper,joint_mask_hands,joint_mask_lower,use_trans,mean_upper,mean_hands,mean_lower,std_upper,std_hands,std_lower,trans_mean,trans_std):
345
  diffusion = create_gaussian_diffusion(use_ddim=args.use_ddim)
346
  args,model,vq_model_upper,vq_model_hands,vq_model_lower,mean_upper,mean_hands,mean_lower,std_upper,std_hands,std_lower,trans_mean,trans_std,vqvae_latent_scale=_warp_create_cuda_model(args,model,mean_upper,mean_hands,mean_lower,std_upper,std_hands,std_lower,trans_mean,trans_std)