update

common.py

@@ -107,44 +107,44 @@ if os.getenv("GRADIO_APP").startswith("imageto3d"):
         os.path.dirname(os.path.abspath(__file__)), "sessions/imageto3d"
     )
     os.makedirs(TMP_DIR, exist_ok=True)
-elif os.getenv("GRADIO_APP").startswith("textto3d"):
-    RBG_REMOVER = RembgRemover()
-    RBG14_REMOVER = BMGG14Remover()
-    if "sam3d" in os.getenv("GRADIO_APP"):
-        PIPELINE = Sam3dInference()
-    else:
-        PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
-            "microsoft/TRELLIS-image-large"
-        )
-        # PIPELINE.cuda()
-    text_model_dir = "weights/Kolors"
-    PIPELINE_IMG_IP = build_text2img_ip_pipeline(text_model_dir, ref_scale=0.3)
-    PIPELINE_IMG = build_text2img_pipeline(text_model_dir)
-    SEG_CHECKER = ImageSegChecker(GPT_CLIENT)
-    GEO_CHECKER = MeshGeoChecker(GPT_CLIENT)
-    AESTHETIC_CHECKER = ImageAestheticChecker()
-    CHECKERS = [GEO_CHECKER, SEG_CHECKER, AESTHETIC_CHECKER]
-    TMP_DIR = os.path.join(
-        os.path.dirname(os.path.abspath(__file__)), "sessions/textto3d"
-    )
-    os.makedirs(TMP_DIR, exist_ok=True)
-elif os.getenv("GRADIO_APP") == "texture_edit":
-    DELIGHT = DelightingModel()
-    IMAGESR_MODEL = ImageRealESRGAN(outscale=4)
-    PIPELINE_IP = build_texture_gen_pipe(
-        base_ckpt_dir="./weights",
-        ip_adapt_scale=0.7,
-        device="cuda",
-    )
-    PIPELINE = build_texture_gen_pipe(
-        base_ckpt_dir="./weights",
-        ip_adapt_scale=0,
-        device="cuda",
-    )
-    TMP_DIR = os.path.join(
-        os.path.dirname(os.path.abspath(__file__)), "sessions/texture_edit"
-    )
-    os.makedirs(TMP_DIR, exist_ok=True)
+# elif os.getenv("GRADIO_APP").startswith("textto3d"):
+#     RBG_REMOVER = RembgRemover()
+#     RBG14_REMOVER = BMGG14Remover()
+#     if "sam3d" in os.getenv("GRADIO_APP"):
+#         PIPELINE = Sam3dInference()
+#     else:
+#         PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
+#             "microsoft/TRELLIS-image-large"
+#         )
+#         # PIPELINE.cuda()
+#     text_model_dir = "weights/Kolors"
+#     PIPELINE_IMG_IP = build_text2img_ip_pipeline(text_model_dir, ref_scale=0.3)
+#     PIPELINE_IMG = build_text2img_pipeline(text_model_dir)
+#     SEG_CHECKER = ImageSegChecker(GPT_CLIENT)
+#     GEO_CHECKER = MeshGeoChecker(GPT_CLIENT)
+#     AESTHETIC_CHECKER = ImageAestheticChecker()
+#     CHECKERS = [GEO_CHECKER, SEG_CHECKER, AESTHETIC_CHECKER]
+#     TMP_DIR = os.path.join(
+#         os.path.dirname(os.path.abspath(__file__)), "sessions/textto3d"
+#     )
+#     os.makedirs(TMP_DIR, exist_ok=True)
+# elif os.getenv("GRADIO_APP") == "texture_edit":
+#     DELIGHT = DelightingModel()
+#     IMAGESR_MODEL = ImageRealESRGAN(outscale=4)
+#     PIPELINE_IP = build_texture_gen_pipe(
+#         base_ckpt_dir="./weights",
+#         ip_adapt_scale=0.7,
+#         device="cuda",
+#     )
+#     PIPELINE = build_texture_gen_pipe(
+#         base_ckpt_dir="./weights",
+#         ip_adapt_scale=0,
+#         device="cuda",
+#     )
+#     TMP_DIR = os.path.join(
+#         os.path.dirname(os.path.abspath(__file__)), "sessions/texture_edit"
+#     )
+#     os.makedirs(TMP_DIR, exist_ok=True)
 
 
 def start_session(req: gr.Request) -> None: