Spaces:
Paused
Paused
Commit
·
5af079f
1
Parent(s):
947dd5a
try to install everything at first
Browse files- models/model_3d_generator.py +8 -8
- 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 |
-
|
124 |
-
|
125 |
-
|
126 |
-
|
127 |
-
|
128 |
-
|
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
|