SkinTokens / requirements.txt
pookiefoof's picture
Public release: SkinTokens · TokenRig demo
9d7cf7f
--extra-index-url https://download.pytorch.org/whl/cu128
--extra-index-url https://pypi.org/simple
# Pinned to match the flash-attn wheel below (cu12torch2.9cxx11abiTRUE).
# Don't bump torch without also bumping the flash-attn URL — they must agree on
# the (cu12 / torch 2.9 / cxx11-abi=TRUE / cp312) tuple.
torch==2.9.1
torchvision==0.24.1
torchaudio==2.9.1
https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.9cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
transformers==4.57.0
diffusers==0.36.0
python-box
einops
omegaconf
pytorch_lightning
lightning
addict
timm
fast-simplification
trimesh
open3d
pyrender
# bpy is NOT listed here. The bpy wheel committed in this repo
# (`bpy-4.5.4rc0-cp312-cp312-manylinux_2_39_x86_64.whl`) is installed at
# runtime by `demo.py` — see `_ensure_bpy_installed()`. Reason: HF Spaces'
# Docker build mounts only `requirements.txt` BEFORE the repo COPY, so the
# wheel path doesn't exist at pip-install time. Public PyPI also has no bpy
# wheel matching this exact build (`rc0` / manylinux_2_39 / cp312).
huggingface_hub
spaces
wandb
numpy==2.2.6
gradio
bottle
tornado
cython