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
|