# 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