Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
3cc12c0
1
Parent(s):
b8ddd5f
add
Browse files
app.py
CHANGED
@@ -333,7 +333,7 @@ class BaseTrainer(object):
|
|
333 |
return result
|
334 |
|
335 |
|
336 |
-
@spaces.GPU(duration=
|
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)
|