unirig_requirements.txt

# UniRig Dependencies (to be installed into Blender's Python 3.11)

# Numpy (UniRig: 1.26.4)
numpy==1.26.4

# PyTorch (Matching main env for simplicity, ensure compatibility with UniRig >=2.3.1)
# Install the version compatible with Python 3.11 and CUDA 12.1
# Example using index-url, adjust if needed
# Note: This will be installed *first* by setup_blender.sh, but listing it here ensures
# pip knows the version constraint when resolving other dependencies later.
torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121
torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu121

# PyTorch Geometric libraries (for CUDA 12.1, Python 3.11)
# Use -f to point pip to the HTML index pages. Pip should find the correct wheels.
# Make sure these URLs correspond to torch 2.3.1 + cu121
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 - check compatibility with Python 3.11 if issues arise)
spconv-cu121

# Dependencies from UniRig's official requirements.txt (excluding bpy and flash_attn)
transformers
python-box
einops
omegaconf
pytorch_lightning # Or just 'lightning' depending on UniRig's imports
lightning         # Might be redundant if pytorch_lightning is used
addict
timm
fast-simplification
flash_attn
trimesh
open3d
pyrender
huggingface_hub
scipy
smplx
chumpy
opencv-python
Pillow
pyquaternion
hydra-core
rich
distinctipy
scikit-image
mediapipe
bpy==4.2