robinwitch commited on
Commit
3cc12c0
·
1 Parent(s): b8ddd5f
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -333,7 +333,7 @@ class BaseTrainer(object):
333
  return result
334
 
335
 
336
- @spaces.GPU(duration=140)
337
  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):
338
  diffusion = create_gaussian_diffusion(use_ddim=args.use_ddim)
339
  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)
 
333
  return result
334
 
335
 
336
+ @spaces.GPU(duration=80)
337
  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):
338
  diffusion = create_gaussian_diffusion(use_ddim=args.use_ddim)
339
  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)