Support zeroGPU

preprocess/humanparsing/run_parsing.py

@@ -47,8 +47,8 @@ def download_model_file(model_path, url):
 class Parsing:
     def __init__(self, gpu_id: int):
         # zeroGPU
-        self.gpu_id = gpu_id
-        torch.cuda.set_device(gpu_id)
+        # self.gpu_id = gpu_id
+        # torch.cuda.set_device(gpu_id)
         
         # Model paths
         base_path = Path(__file__).absolute().parents[2].absolute()
@@ -76,6 +76,6 @@ class Parsing:
 
     def __call__(self, input_image):
         # zeroGPU
-        torch.cuda.set_device(self.gpu_id)
+        # torch.cuda.set_device(self.gpu_id)
         parsed_image, face_mask = onnx_inference(self.session, self.lip_session, input_image)
         return parsed_image, face_mask

preprocess/openpose/run_openpose.py

@@ -28,13 +28,13 @@ import pdb
 class OpenPose:
     def __init__(self, gpu_id: int):
         # zeroGPU
-        self.gpu_id = gpu_id
-        torch.cuda.set_device(gpu_id)
+        # self.gpu_id = gpu_id
+        # torch.cuda.set_device(gpu_id)
         self.preprocessor = OpenposeDetector()
 
     def __call__(self, input_image, resolution=384):
         # zeroGPU
-        torch.cuda.set_device(self.gpu_id)
+        # torch.cuda.set_device(self.gpu_id)
         if isinstance(input_image, Image.Image):
             input_image = np.asarray(input_image)
         elif type(input_image) == str: