open3dforge / requirements.txt
Reverb's picture
Switch to torch 2.9.1 + runtime nvdiffrast install
ec19943
Raw
History Blame Contribute Delete
2.4 kB
# Open3DForge — Dependencies
--extra-index-url https://download.pytorch.org/whl/cu124
# ===== Milestone 1: Foundation =====
gradio>=5.0
spaces # HF ZeroGPU decorator
numpy
pillow
# ===== Milestone 2: Stage 1 Generation =====
# ZeroGPU supported torch versions: 2.8.0, 2.9.1, 2.10.0, 2.11.0.
# 2.8.0 compiled with CUDA 12.8; ZeroGPU machine has CUDA 13.0 toolkit —
# any C++ extension build (incl. nvdiffrast) fails with a version mismatch.
# 2.9.1 is the first version likely compiled with CUDA 13.0, so nvdiffrast
# source build (runtime, --no-build-isolation) should succeed.
gradio_client # call TRELLIS.2 Space
torch==2.9.1
torchvision==0.24.1
imageio==2.37.2
imageio-ffmpeg==0.6.0
tqdm==4.67.1
easydict==1.13
opencv-python-headless==4.12.0.88
trimesh==4.10.1
transformers==4.57.3
zstandard==0.25.0
kornia==0.8.2
timm==1.0.22
git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
# Hunyuan3D-2.1 dependencies
diffusers>=0.30.0
accelerate
einops
omegaconf
# Background removal (used before shape generation)
rembg
onnxruntime
# ===== Milestone 3: Stage 2A-2C Mesh Cleanup =====
trimesh[easy]
pymeshfix
# pymeshlab is required by hy3dshape/postprocessors.py (imported at module
# level, so must be installable). Its OpenGL-dependent filter plugins print
# warnings on ZeroGPU headless but do NOT crash on import. Our own pipeline
# stages use fast-simplification for decimation/LODs to avoid those filters.
pymeshlab
fast-simplification
# ===== Milestone 4: Stage 2D-2E UV + Symmetry =====
xatlas
scipy # normal map dilation
# ===== Milestone 5: Stage 2F Normal Baking =====
# nvdiffrast is installed at app startup via _install_nvdiffrast() in app.py.
# Reason: pip build isolation hides torch, causing "Cannot compile" error.
# Runtime install with --no-build-isolation succeeds because torch is already
# in the environment. Requires torch 2.9.1+ (CUDA 13.0) to match the machine.
coacd==1.0.4 # Milestone 8 collision
# ===== Milestone 4: Stage 2 GPU baking =====
# nvdiffrast (install from PyPI — compatible with current torch)
# ===== Milestone 5: Stage 2 SDXL inpainting =====
# diffusers
# accelerate
# safetensors
# ===== Milestone 7: Stage 3 Auto-rigging =====
# UniRig — via gradio_client (same ABI strategy as TRELLIS.2)
# ===== Milestone 8: Export =====
# pygltflib