update serve

model/model_worker.py

@@ -63,60 +63,60 @@ class HuggingfaceApiWorker(BaseModelWorker):
             model_api,
         )
 
-class PointE_Worker(BaseModelWorker):
-    def __init__(self, 
-                 model_name: str, 
-                 i2s_model: bool, 
-                 online_model: bool, 
-                 model_api: str):
-        super().__init__(model_name, i2s_model, online_model, model_api)
-
-class TriplaneGaussian(BaseModelWorker):
-    def __init__(self, model_name: str, i2s_model: bool, online_model: bool, model_api: str = None):
-        super().__init__(model_name, i2s_model, online_model, model_api)
-
-
-class LGM_Worker(BaseModelWorker):
-    def __init__(self, 
-                 model_name: str, 
-                 i2s_model: bool, 
-                 online_model: bool, 
-                 model_api: str = "camenduru/lgm:d2870893aa115773465a823fe70fd446673604189843f39a99642dd9171e05e2",
-    ):
-        super().__init__(model_name, i2s_model, online_model, model_api)
-        self.model_client = replicate.Client(api_token=REPLICATE_API_TOKEN)
+# class PointE_Worker(BaseModelWorker):
+#     def __init__(self, 
+#                  model_name: str, 
+#                  i2s_model: bool, 
+#                  online_model: bool, 
+#                  model_api: str):
+#         super().__init__(model_name, i2s_model, online_model, model_api)
+
+# class TriplaneGaussian(BaseModelWorker):
+#     def __init__(self, model_name: str, i2s_model: bool, online_model: bool, model_api: str = None):
+#         super().__init__(model_name, i2s_model, online_model, model_api)
+
+
+# class LGM_Worker(BaseModelWorker):
+#     def __init__(self, 
+#                  model_name: str, 
+#                  i2s_model: bool, 
+#                  online_model: bool, 
+#                  model_api: str = "camenduru/lgm:d2870893aa115773465a823fe70fd446673604189843f39a99642dd9171e05e2",
+#     ):
+#         super().__init__(model_name, i2s_model, online_model, model_api)
+#         self.model_client = replicate.Client(api_token=REPLICATE_API_TOKEN)
     
-    def inference(self, image):
+#     def inference(self, image):
         
-        output = self.model_client.run(
-            self.model_api,
-            input={"input_image": image}
-        )
-        #=> .mp4 .ply
-        return output[1]
-
-    def render(self, shape):
-        mesh = Gau2Mesh_client.run(shape)
-
-        path_normal = ""
-        cmd_normal = f"python -m ..kiuikit.kiui.render {mesh} --save {path_normal} \
-            --wogui --H 512 --W 512 --radius 3 --elevation 0 --num_azimuth 40 --front_dir='+z' --mode normal"
-        subprocess.run(cmd_normal, shell=True, check=True)
-
-        path_rgb = ""
-        cmd_rgb = f"python -m ..kiuikit.kiui.render {mesh} --save {path_rgb} \
-            --wogui --H 512 --W 512 --radius 3 --elevation 0 --num_azimuth 40 --front_dir='+z' --mode rgb"
-        subprocess.run(cmd_rgb, shell=True, check=True)
+#         output = self.model_client.run(
+#             self.model_api,
+#             input={"input_image": image}
+#         )
+#         #=> .mp4 .ply
+#         return output[1]
+
+#     def render(self, shape):
+#         mesh = Gau2Mesh_client.run(shape)
+
+#         path_normal = ""
+#         cmd_normal = f"python -m ..kiuikit.kiui.render {mesh} --save {path_normal} \
+#             --wogui --H 512 --W 512 --radius 3 --elevation 0 --num_azimuth 40 --front_dir='+z' --mode normal"
+#         subprocess.run(cmd_normal, shell=True, check=True)
+
+#         path_rgb = ""
+#         cmd_rgb = f"python -m ..kiuikit.kiui.render {mesh} --save {path_rgb} \
+#             --wogui --H 512 --W 512 --radius 3 --elevation 0 --num_azimuth 40 --front_dir='+z' --mode rgb"
+#         subprocess.run(cmd_rgb, shell=True, check=True)
         
-        return path_normal, path_rgb
-
-class V3D_Worker(BaseModelWorker):
-    def __init__(self, 
-                model_name: str, 
-                i2s_model: bool, 
-                online_model: bool, 
-                model_api: str = None):
-        super().__init__(model_name, i2s_model, online_model, model_api)
+#         return path_normal, path_rgb
+
+# class V3D_Worker(BaseModelWorker):
+#     def __init__(self, 
+#                 model_name: str, 
+#                 i2s_model: bool, 
+#                 online_model: bool, 
+#                 model_api: str = None):
+#         super().__init__(model_name, i2s_model, online_model, model_api)
 
 
 # model = 'LGM'