add

app.py

@@ -267,7 +267,7 @@ class BaseTrainer(object):
             for its, batch_data in enumerate(self.test_loader):
                 # loaded_data = self._load_data(batch_data)    
                 # net_out = self._g_test(loaded_data)
-                net_out = _warp(self.args,self.model, batch_data,self.joints,self.joint_mask_upper,self.joint_mask_hands,self.joint_mask_lower,self.use_trans)
+                net_out = _warp(self.args,self.model, batch_data,self.joints,self.joint_mask_upper,self.joint_mask_hands,self.joint_mask_lower,self.use_trans,self.mean_upper,self.mean_hands,self.mean_lower,self.std_upper,self.std_hands,self.std_lower,self.trans_mean,self.trans_std)
                 tar_pose = net_out['tar_pose']
                 rec_pose = net_out['rec_pose']
                 tar_exps = net_out['tar_exps']
@@ -334,9 +334,9 @@ class BaseTrainer(object):
 
 
 @spaces.GPU(duration=149)
-def _warp(args,model, batch_data,joints,joint_mask_upper,joint_mask_hands,joint_mask_lower,use_trans):
+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):
     diffusion = create_gaussian_diffusion(use_ddim=args.use_ddim)
-    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)
+    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)
     
     
     loaded_data = _warp_load_data(
@@ -635,7 +635,7 @@ def _warp_load_data(dict_data,joints,joint_mask_upper,joint_mask_hands,joint_mas
     }
 
 
-def _warp_create_cuda_model(args,model):
+def _warp_create_cuda_model(args,model,mean_upper,mean_hands,mean_lower,std_upper,std_hands,std_lower,trans_mean,trans_std):
     args = args
     other_tools.load_checkpoints(model, args.test_ckpt, args.g_name)
     args.num_quantizers = 6