NeAR / requirements.txt
luh1124's picture
app: CPU preload Hunyuan+NeAR, cuda move in GPU callbacks; drop gsplat warmup
c98c836
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.4.0
torchvision==0.19.0
torchaudio==2.4.0
xformers==0.0.27.post2
huggingface_hub>=0.26.0
gradio[oauth,mcp]==6.9.0
spaces
websockets>=10.4
simple_ocio
--find-links https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.4.0_cu124.html
kaolin
# NeAR demo / inference (see NeAR setup.sh --basic --demo)
pillow
imageio
imageio-ffmpeg
tqdm
easydict
opencv-python-headless
scipy
ninja
rembg
onnxruntime
trimesh
open3d
xatlas
pyvista
pymeshfix
igraph
transformers==4.57.6
pyexr
# gsplat
pyyaml
# hy3dshape (vendored Hunyuan3D-2.1)
pybind11
diffusers
einops
timm
kornia
omegaconf
accelerate
fastapi
uvicorn>=0.14.0
pymeshlab
pygltflib
# Sparse conv: CUDA 12.x wheel (use with PyTorch cu128 / driver 12.x).
spconv-cu120
# vox2seq: do not list ./vox2seq here — HF mounts requirements.txt before the repo tree exists, so path deps fail.
# serialized_attn falls back to trellis.modules.sparse.attention.vox2seq_pytorch (pure torch). Optional: pip install ./vox2seq locally for CUDA _C.
git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
# nvdiffrast: custom wheel (torch 2.8/cu128 ABI) — pip needs /resolve/, not /blob/
# https://huggingface.co/luh0502/near-wheels/resolve/main/nvdiffrast-0.4.0-cp310-cp310-linux_x86_64.whl
https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
# gsplat==1.5.3
# https://huggingface.co/luh0502/near-wheels/resolve/main/gsplat-1.5.3-cp310-cp310-linux_x86_64.whl
https://huggingface.co/luh0502/near-wheels/resolve/main/gsplat-1.5.3+pt24cu124-cp310-cp310-linux_x86_64.whl
# https://docs.gsplat.studio/whl/gsplat/gsplat-1.5.3+pt24cu124-cp310-cp310-linux_x86_64.whl