BladeSzaSza commited on
Commit
5af079f
·
1 Parent(s): 947dd5a

try to install everything at first

Browse files
Files changed (2) hide show
  1. models/model_3d_generator.py +8 -8
  2. requirements.txt +69 -1
models/model_3d_generator.py CHANGED
@@ -118,14 +118,14 @@ class Hunyuan3DGenerator:
118
  logger.info("🔧 Running Hunyuan3D setup scripts with detailed logging...")
119
 
120
  # 1. Install requirements from the model's specific requirements file
121
- requirements_path = os.path.join(self.model_path, 'requirements_hunyuan3d.txt')
122
- if os.path.exists(requirements_path):
123
- pip_command = [
124
- sys.executable, '-m', 'pip', 'install', '-r', requirements_path,
125
- '--extra-index-url', 'https://mirrors.cloud.tencent.com/pypi/simple/',
126
- '--extra-index-url', 'https://mirrors.aliyun.com/pypi/simple'
127
- ]
128
- run_setup_command(pip_command, cwd=self.model_path)
129
 
130
  # 2. Install custom rasterizer dependencies (torch)
131
  # logger.info("Installing torch, torchvision, torchaudio...")
 
118
  logger.info("🔧 Running Hunyuan3D setup scripts with detailed logging...")
119
 
120
  # 1. Install requirements from the model's specific requirements file
121
+ # requirements_path = os.path.join(self.model_path, 'requirements_hunyuan3d.txt')
122
+ # if os.path.exists(requirements_path):
123
+ # pip_command = [
124
+ # sys.executable, '-m', 'pip', 'install', '-r', requirements_path,
125
+ # '--extra-index-url', 'https://mirrors.cloud.tencent.com/pypi/simple/',
126
+ # '--extra-index-url', 'https://mirrors.aliyun.com/pypi/simple'
127
+ # ]
128
+ # run_setup_command(pip_command, cwd=self.model_path)
129
 
130
  # 2. Install custom rasterizer dependencies (torch)
131
  # logger.info("Installing torch, torchvision, torchaudio...")
requirements.txt CHANGED
@@ -45,4 +45,72 @@ aiofiles>=23.2.1
45
 
46
  # Optional optimizations
47
  # onnxruntime-gpu>=1.16.0 # For ONNX model support
48
- # xformers>=0.0.23 # For memory-efficient attention
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
 
46
  # Optional optimizations
47
  # onnxruntime-gpu>=1.16.0 # For ONNX model support
48
+ # xformers>=0.0.23 # For memory-efficient attention
49
+
50
+ --extra-index-url https://mirrors.cloud.tencent.com/pypi/simple/
51
+ --extra-index-url https://mirrors.aliyun.com/pypi/simple
52
+
53
+ # Build Tools
54
+ ninja==1.11.1.1
55
+ pybind11==2.13.4
56
+
57
+ # Core ML/Deep Learning
58
+ transformers==4.46.0
59
+ diffusers==0.30.0
60
+ accelerate==1.1.1
61
+ pytorch-lightning==1.9.5
62
+ huggingface-hub==0.30.2
63
+ safetensors==0.4.4
64
+
65
+ # Scientific Computing
66
+ numpy==1.24.4
67
+ scipy==1.14.1
68
+ einops==0.8.0
69
+ pandas==2.2.2
70
+
71
+ # Computer Vision & Image Processing
72
+ opencv-python==4.10.0.84
73
+ imageio==2.36.0
74
+ scikit-image==0.24.0
75
+ rembg==2.0.65
76
+ realesrgan==0.3.0
77
+ tb_nightly==2.18.0a20240726
78
+ basicsr==1.4.2
79
+
80
+ # 3D Mesh Processing
81
+ trimesh==4.4.7
82
+ pymeshlab==2022.2.post3
83
+ pygltflib==1.16.3
84
+ xatlas==0.0.9
85
+ open3d==0.18.0
86
+
87
+ # Configuration Management
88
+ omegaconf==2.3.0
89
+ pyyaml==6.0.2
90
+ configargparse==1.7
91
+
92
+ # Web Framework (for demo)
93
+ gradio==5.33.0
94
+ fastapi==0.115.12
95
+ uvicorn==0.34.3
96
+
97
+ # Utilities
98
+ tqdm==4.66.5
99
+ psutil==6.0.0
100
+
101
+ # GPU Computing (requires CUDA)
102
+ cupy-cuda12x==13.4.1
103
+
104
+ # Blender
105
+ bpy==4.0
106
+
107
+ # ONNX Runtime
108
+ onnxruntime==1.16.3
109
+ torchmetrics==1.6.0
110
+
111
+ pydantic==2.10.6
112
+
113
+ timm
114
+ pythreejs
115
+ torchdiffeq
116
+ deepspeed