Update model.py

model.py

@@ -26,7 +26,7 @@ class in2INModel(PreTrainedModel):
         self.model.eval()
         batch = OrderedDict({})
 
-        batch["motion_lens"] = torch.zeros(1,1).long().cuda()
+        batch["motion_lens"] = torch.zeros(1,1).long()
         batch["prompt_interaction"] = prompt_interaction
 
         if self.mode != "individual":
@@ -53,14 +53,6 @@ class in2INModel(PreTrainedModel):
             batch["text_individual2"] = [prompt_individual2]
 
         batch = self.model.forward_test(batch)
-
-        if self.mode == "individual":
-            motion_output = batch["output"][0].reshape(-1, 262)
-            motion_output = self.normalizer.backward(motion_output.cpu().detach().numpy())
-            joints3d = motion_output[:,:22*3].reshape(-1,22,3)
-            joints3d = gaussian_filter1d(joints3d, 1, axis=0, mode='nearest')
-            return joints3d
-
         motion_output_both = batch["output"][0].reshape(batch["output"][0].shape[0], 2, -1)
         motion_output_both = self.normalizer.backward(motion_output_both.cpu().detach().numpy())