fixed the requiriments

models/model_3d_generator.py

@@ -127,7 +127,12 @@ class Hunyuan3DGenerator:
                     ]
                     run_setup_command(pip_command, cwd=self.model_path)
 
-                # 2. Install custom rasterizer
+                # 2. Install custom rasterizer dependencies (torch)
+                logger.info("Installing torch, torchvision, torchaudio...")
+                pip_command_torch = [sys.executable, '-m', 'pip', 'install', 'torch==2.5.1', 'torchvision==0.20.1', 'torchaudio==2.5.1', '--index-url', 'https://download.pytorch.org/whl/cu124']
+                run_setup_command(pip_command_torch, cwd=self.model_path)
+
+                # 3. Install custom rasterizer
                 rasterizer_path = os.path.join(self.model_path, 'hy3dpaint', 'packages', 'custom_rasterizer')
                 if os.path.exists(rasterizer_path):
                     pip_command_rasterizer = [sys.executable, '-m', 'pip', 'install', '-e', '.']

requirements.txt

@@ -3,9 +3,9 @@ gradio>=4.16.0
 spaces>=0.19.0
 
 # AI/ML frameworks
-torch>=2.1.0
-torchvision>=0.16.0
-torchaudio>=2.1.0
+torch>=2.5.1
+torchvision>=0.20.1
+torchaudio>=2.5.1
 transformers>=4.36.0
 diffusers>=0.24.0
 accelerate>=0.25.0

requirements_hunyuan3d.txt

@@ -1,37 +1,67 @@
-# Additional requirements for Hunyuan3D-2.1 support
-# Install these in addition to the main requirements.txt
-
-# Hunyuan3D specific dependencies
-# Note: The hy3dshape and hy3dpaint modules need to be installed from the Hunyuan3D repository
-# git clone https://huggingface.co/spaces/tencent/Hunyuan3D-2.1
-# Then copy the hy3dshape and hy3dpaint folders to your project
-
-# Core dependencies that Hunyuan3D uses
-trimesh>=3.23.5
-pygltflib>=1.16.1
-scikit-image>=0.22.0
-opencv-python>=4.8.0
-Pillow>=10.0.0
-numpy>=1.24.0
-torch>=2.0.0
-torchvision>=0.15.0
-transformers>=4.35.0
-diffusers>=0.25.0
-accelerate>=0.25.0
-safetensors>=0.4.0
-huggingface-hub>=0.19.0
-spaces>=0.19.0
-
-# For background removal
-rembg>=2.0.50
-
-# For 3D visualization and processing
-pymeshlab>=2022.2.post3
-open3d>=0.17.0
-
-# Optional: For mesh optimization
-xatlas>=0.0.7
-pyvista>=0.42.0
-
-# Note: Some dependencies might require specific versions or CUDA compatibility
-# Adjust versions based on your system configuration 
+--extra-index-url https://mirrors.cloud.tencent.com/pypi/simple/
+--extra-index-url https://mirrors.aliyun.com/pypi/simple
+
+# Build Tools
+ninja==1.11.1.1
+pybind11==2.13.4
+
+# Core ML/Deep Learning
+transformers==4.46.0
+diffusers==0.30.0
+accelerate==1.1.1
+pytorch-lightning==1.9.5
+huggingface-hub==0.30.2
+safetensors==0.4.4
+
+# Scientific Computing
+numpy==1.24.4
+scipy==1.14.1
+einops==0.8.0
+pandas==2.2.2
+
+# Computer Vision & Image Processing
+opencv-python==4.10.0.84
+imageio==2.36.0
+scikit-image==0.24.0
+rembg==2.0.65
+realesrgan==0.3.0
+tb_nightly==2.18.0a20240726
+basicsr==1.4.2
+
+# 3D Mesh Processing
+trimesh==4.4.7
+pymeshlab==2022.2.post3
+pygltflib==1.16.3
+xatlas==0.0.9
+open3d==0.18.0
+
+# Configuration Management
+omegaconf==2.3.0
+pyyaml==6.0.2
+configargparse==1.7
+
+# Web Framework (for demo)
+gradio==5.33.0
+fastapi==0.115.12
+uvicorn==0.34.3
+
+# Utilities
+tqdm==4.66.5
+psutil==6.0.0
+
+# GPU Computing (requires CUDA)
+cupy-cuda12x==13.4.1
+
+# Blender
+bpy==4.0
+
+# ONNX Runtime
+onnxruntime==1.16.3
+torchmetrics==1.6.0
+
+pydantic==2.10.6
+
+timm
+pythreejs
+torchdiffeq
+deepspeed