Update gradio_app.py

gradio_app.py

@@ -1,12 +1,15 @@
 # Hunyuan 3D is licensed under the TENCENT HUNYUAN NON-COMMERCIAL LICENSE AGREEMENT
 # except for the third-party components listed below.
 
+import importlib
 import os
 import random
 import shutil
+import site
 import subprocess
 import sys
 import time
+import urllib.request
 import uuid
 from glob import glob
 from pathlib import Path
@@ -23,22 +26,36 @@ from hy3dgen.shapegen.utils import logger
 
 MAX_SEED = 1e7
 
+texgen_worker = None
+texgen_load_error = None
+
+
+def reset_diffusers_module_cache():
+    candidates = [
+        Path(os.environ.get("HF_MODULES_CACHE", "/tmp/hf_modules")) / "diffusers_modules" / "local",
+        Path.home() / ".cache" / "huggingface" / "modules" / "diffusers_modules" / "local",
+    ]
+
+    for cache_path in candidates:
+        try:
+            if cache_path.exists():
+                print(f"Removing stale Diffusers dynamic module cache: {cache_path}")
+                shutil.rmtree(cache_path, ignore_errors=True)
+        except Exception as exc:
+            print(f"Could not remove Diffusers module cache {cache_path}: {exc}")
+
+    importlib.invalidate_caches()
+
 
 # Rebuild texture native extensions against the currently installed PyTorch.
 # This avoids ABI mismatch from the bundled prebuilt wheel.
 if True:
-    import importlib
-    import os
-    import site
-    import spaces
-    import subprocess
-    import sys
-    import urllib.request
-    from pathlib import Path
-
     os.environ.setdefault("MAX_JOBS", "1")
     os.environ.setdefault("TORCH_CUDA_ARCH_LIST", "8.9")
     os.environ.setdefault("CUDAARCHS", "89")
+    os.environ.setdefault("HF_MODULES_CACHE", "/tmp/hf_modules")
+
+    reset_diffusers_module_cache()
 
     rasterizer_root = Path("/home/user/app/hy3dgen/texgen/custom_rasterizer")
     kernel_dir = rasterizer_root / "lib" / "custom_rasterizer_kernel"
@@ -89,7 +106,6 @@ if True:
         check=True,
     )
 
-    # Refresh current Python process so imports installed during startup are visible.
     for site_dir in site.getsitepackages():
         if site_dir not in sys.path:
             sys.path.append(site_dir)
@@ -167,10 +183,6 @@ if True:
     importlib.invalidate_caches()
 
 
-
-
-
-
 def get_example_img_list():
     print("Loading example img list ...")
     return sorted(glob("./assets/example_images/**/*.png", recursive=True))
@@ -268,6 +280,47 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
     """
 
 
+def get_texgen_worker():
+    global texgen_worker, texgen_load_error
+
+    if texgen_worker is not None:
+        return texgen_worker
+
+    if args.disable_tex:
+        raise gr.Error("Texture generation is disabled with --disable_tex.")
+
+    last_error = None
+
+    for attempt in range(2):
+        try:
+            if attempt > 0:
+                print("Retrying texture generator after cleaning Diffusers module cache...")
+                reset_diffusers_module_cache()
+
+            from hy3dgen.texgen import Hunyuan3DPaintPipeline
+
+            print("Loading texture generator on demand...")
+            texgen_worker = Hunyuan3DPaintPipeline.from_pretrained(args.texgen_model_path)
+
+            if args.low_vram_mode:
+                texgen_worker.enable_model_cpu_offload()
+
+            texgen_load_error = None
+            print("Texture generator loaded successfully.")
+            return texgen_worker
+
+        except Exception as exc:
+            import traceback
+
+            last_error = exc
+            texgen_load_error = traceback.format_exc()
+            print(texgen_load_error)
+
+            if "diffusers_modules.local.modules" not in texgen_load_error and attempt == 0:
+                break
+
+    raise gr.Error(f"Texture generator failed to load: {last_error}")
+
 
 def _gen_shape(
     caption=None,
@@ -375,7 +428,6 @@ def _gen_shape(
     return mesh, main_image, save_folder, stats, seed
 
 
-
 def generation_all(
     caption=None,
     image=None,
@@ -417,7 +469,8 @@ def generation_all(
     stats["time"]["face reduction"] = time.time() - tmp_time
 
     tmp_time = time.time()
-    textured_mesh = texgen_worker(mesh, image)
+    texgen = get_texgen_worker()
+    textured_mesh = texgen(mesh, image)
     logger.info("---Texture Generation takes %s seconds ---" % (time.time() - tmp_time))
     stats["time"]["texture generation"] = time.time() - tmp_time
     stats["time"]["total"] = time.time() - start_time_0
@@ -444,7 +497,6 @@ def generation_all(
     )
 
 
-
 def shape_generation(
     caption=None,
     image=None,
@@ -714,7 +766,7 @@ def build_app():
         gr.HTML(f"""
         <div align="center">
         Activated Model - Shape Generation ({args.model_path}/{args.subfolder}) ;
-        Texture Generation ({'Hunyuan3D-2' if HAS_TEXTUREGEN else 'Unavailable'})
+        Texture Generation ({'Hunyuan3D-2 loads on demand' if HAS_TEXTUREGEN else 'Unavailable'})
         </div>
         """)
 
@@ -722,8 +774,7 @@ def build_app():
             gr.HTML("""
             <div style="margin-top: 5px;" align="center">
                 <b>Warning: </b>
-                Texture synthesis is disabled due to missing requirements.
-                Check the Space logs for the exact Python traceback.
+                Texture synthesis is disabled with --disable_tex.
             </div>
             """)
 
@@ -920,25 +971,7 @@ if __name__ == "__main__":
 
     SUPPORTED_FORMATS = ["glb", "obj", "ply", "stl"]
 
-    HAS_TEXTUREGEN = False
-    if not args.disable_tex:
-        try:
-            from hy3dgen.texgen import Hunyuan3DPaintPipeline
-
-            texgen_worker = Hunyuan3DPaintPipeline.from_pretrained(args.texgen_model_path)
-
-            if args.low_vram_mode:
-                texgen_worker.enable_model_cpu_offload()
-
-            HAS_TEXTUREGEN = True
-        except Exception as e:
-            import traceback
-
-            traceback.print_exc()
-            print(e)
-            print("Failed to load texture generator.")
-            print("Please try to install requirements by following README.md")
-            HAS_TEXTUREGEN = False
+    HAS_TEXTUREGEN = not args.disable_tex
 
     HAS_T2I = True
     if args.enable_t23d:
@@ -986,4 +1019,3 @@ if __name__ == "__main__":
     app = gr.mount_gradio_app(app, demo, path="/")
 
     uvicorn.run(app, host=args.host, port=args.port)
-