Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
88abf96
1
Parent(s):
a30f1f1
add
Browse files
app.py
CHANGED
@@ -340,7 +340,7 @@ class BaseTrainer(object):
|
|
340 |
return result
|
341 |
|
342 |
|
343 |
-
@spaces.GPU(duration=
|
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)
|