着手修改依赖版本

gradio_app.py

@@ -683,7 +683,8 @@ if __name__ == '__main__':
     if not args.disable_tex:
         try:
             from hy3dgen.texgen import Hunyuan3DPaintPipeline
-
+            print("着手加载贴图Pipeline")
+            print(args.texgen_model_path)
             texgen_worker = Hunyuan3DPaintPipeline.from_pretrained(args.texgen_model_path)
             if args.low_vram_mode:
                 texgen_worker.enable_model_cpu_offload()
@@ -713,6 +714,9 @@ if __name__ == '__main__':
     from hy3dgen.rembg import BackgroundRemover
 
     rmbg_worker = BackgroundRemover()
+    print("着手加载模型Pipeline")
+    print(args.model_path)
+    print(args.subfolder)
     i23d_worker = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(
         args.model_path,
         subfolder=args.subfolder,
@@ -742,4 +746,4 @@ if __name__ == '__main__':
         torch.cuda.empty_cache()
     demo = build_app()
     app = gr.mount_gradio_app(app, demo, path="/")
-    uvicorn.run(app, host=args.host, port=args.port)
+    uvicorn.run(app, host=args.host, port=args.port)

hy3dgen/texgen/pipelines.py

@@ -19,6 +19,7 @@ import os
 import torch
 from PIL import Image
 from typing import Union, Optional
+from pathlib import Path
 
 from .differentiable_renderer.mesh_render import MeshRender
 from .utils.dehighlight_utils import Light_Shadow_Remover
@@ -50,30 +51,59 @@ class Hunyuan3DPaintPipeline:
     @classmethod
     def from_pretrained(cls, model_path):
         original_model_path = model_path
+        print(f"原始路径 original_model_path: {model_path}")
         if not os.path.exists(model_path):
+            print(f"存在原始路径: {model_path}")
             # try local path
             base_dir = os.environ.get('HY3DGEN_MODELS', '~/.cache/hy3dgen')
             model_path = os.path.expanduser(os.path.join(base_dir, model_path))
 
+            print(f"基础路径 base_dir: {base_dir}")
+            print(f"模型路径 model_path: {model_path}")
+
             delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')
             multiview_model_path = os.path.join(model_path, 'hunyuan3d-paint-v2-0')
 
+            print(f"路径 delight_model_path: {delight_model_path}")
+            print(f"路径 multiview_model_path: {multiview_model_path}")
+
             if not os.path.exists(delight_model_path) or not os.path.exists(multiview_model_path):
                 try:
                     import huggingface_hub
                     # download from huggingface
                     model_path = huggingface_hub.snapshot_download(repo_id=original_model_path,
                                                                    allow_patterns=["hunyuan3d-delight-v2-0/*"])
+                    print(f"下载的 model_path 1 : {model_path}")
+                    snapshot_path = Path(model_path)
+                    for path in snapshot_path.rglob("*"):
+                        print(path.relative_to(snapshot_path))
                     model_path = huggingface_hub.snapshot_download(repo_id=original_model_path,
                                                                    allow_patterns=["hunyuan3d-paint-v2-0/*"])
+                    print(f"下载的 model_path 2 : {model_path}")
+                    snapshot_path = Path(model_path)
+                    for path in snapshot_path.rglob("*"):
+                        print(path.relative_to(snapshot_path))
+
+                    
                     delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')
                     multiview_model_path = os.path.join(model_path, 'hunyuan3d-paint-v2-0')
+                    
+                    print(f"路径 delight_model_path : {delight_model_path}")
+                    print(f"路径 multiview_model_path : {multiview_model_path}")
+                    print(f"路径 delight_model_path 是否存在: {os.path.exists(delight_model_path)}")
+                    print(f"路径 multiview_model_path 是否存在: {os.path.exists(multiview_model_path)}")
+
                     return cls(Hunyuan3DTexGenConfig(delight_model_path, multiview_model_path))
-                except ImportError:
-                    logger.warning(
-                        "You need to install HuggingFace Hub to load models from the hub."
-                    )
-                    raise RuntimeError(f"Model path {model_path} not found")
+                except Exception as e:
+                    print("构造 Hunyuan3DPaintPipeline 实例时出错：", e)
+                    import traceback
+                    traceback.print_exc()
+                    raise
+                # except ImportError:
+                #     logger.warning(
+                #         "You need to install HuggingFace Hub to load models from the hub."
+                #     )
+                #     raise RuntimeError(f"Model path {model_path} not found")
             else:
                 return cls(Hunyuan3DTexGenConfig(delight_model_path, multiview_model_path))
 
@@ -224,4 +254,4 @@ class Hunyuan3DPaintPipeline:
         self.render.set_texture(texture)
         textured_mesh = self.render.save_mesh()
 
-        return textured_mesh
+        return textured_mesh

requirements.txt

@@ -6,9 +6,9 @@ diffusers
 einops
 opencv-python
 numpy
-torch
+torch==2.4.0
 transformers
-torchvision
+torchvision==0.19.0
 #taming-transformers-rom1504
 #ConfigArgParse
 #ipdb
@@ -39,4 +39,3 @@ rembg
 onnxruntime
 #gevent
 #geventhttpclient
-