fix(texture): lazy load IP adapter pipeline

common.py

@@ -138,11 +138,7 @@ elif os.getenv("GRADIO_APP").startswith("textto3d"):
 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_IP = None
     PIPELINE = build_texture_gen_pipe(
         base_ckpt_dir="./weights",
         ip_adapt_scale=0,
@@ -657,7 +653,7 @@ def generate_condition(mesh_path: str, req: gr.Request, uuid: str = "sample"):
     return None, None, None
 
 
-@spaces.GPU
+@spaces.GPU(duration=240)
 def generate_texture_mvimages(
     prompt: str,
     controlnet_cond_scale: float = 0.55,
@@ -671,9 +667,21 @@ def generate_texture_mvimages(
     sub_idxs: tuple[tuple[int]] = ((0, 1, 2), (3, 4, 5)),
     req: gr.Request = None,
 ) -> list[str]:
+    global PIPELINE_IP
+
     output_root = os.path.join(TMP_DIR, str(req.session_hash))
     use_ip_adapter = True if ip_img_path and ip_adapt_scale > 0 else False
-    PIPELINE_IP.set_ip_adapter_scale([ip_adapt_scale])
+    pipeline = PIPELINE
+    if use_ip_adapter:
+        if PIPELINE_IP is None:
+            PIPELINE_IP = build_texture_gen_pipe(
+                base_ckpt_dir="./weights",
+                ip_adapt_scale=ip_adapt_scale,
+                device="cuda",
+            )
+        PIPELINE_IP.set_ip_adapter_scale([ip_adapt_scale])
+        pipeline = PIPELINE_IP
+
     img_save_paths = infer_pipe(
         index_file=f"{output_root}/condition/index.json",
         controlnet_cond_scale=controlnet_cond_scale,
@@ -686,7 +694,7 @@ def generate_texture_mvimages(
         prompt=prompt,
         save_dir=f"{output_root}/multi_view",
         sub_idxs=sub_idxs,
-        pipeline=PIPELINE_IP if use_ip_adapter else PIPELINE,
+        pipeline=pipeline,
         seed=seed,
     )
 

requirements.txt

@@ -1,4 +1,4 @@
---extra-index-url https://download.pytorch.org/whl/cu126
+--extra-index-url https://download.pytorch.org/whl/cu128
 
 setuptools==80.10.2
 wheel