requirements.txt

# PyTorch (UniRig: >=2.3.1)
torch>=2.3.1
torchvision>=0.18.1 # Compatible with torch 2.3.1

# PyTorch Geometric libraries (for CUDA 12.1, matching PyTorch 2.3.1)
torch-scatter -f https://data.pyg.org/whl/torch-2.3.1+cu121.html
torch-cluster -f https://data.pyg.org/whl/torch-2.3.1+cu121.html

# Spconv (for CUDA 12.1)
spconv-cu121

# Numpy (UniRig: 1.26.4)
numpy==1.26.4

# Gradio (Updated to 5.29.0)
gradio>=5.29.0,<5.30.0 

spaces

# Dependencies from UniRig's official requirements.txt
transformers
python-box
einops
omegaconf
pytorch_lightning
lightning
addict
timm
fast-simplification
# Replaced flash_attn with pre-built wheel
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4/flash_attn-2.7.4+cu121torch2.3.1-cp310-cp310-linux_x86_64.whl
trimesh
open3d
pyrender
huggingface_hub
scipy
smplx
chumpy
opencv-python 
Pillow
pyquaternion
hydra-core      
rich            
distinctipy
scikit-image
mediapipe