HexaGrid / requirements.txt
Surn's picture
Ensure diff_gaussian_rasterization is installed at runtime
857fd4e
Raw
History Blame Contribute Delete
5.18 kB
open3d
git+https://github.com/huggingface/diffusers.git
#diffusers[torch]==0.32.2
transformers
accelerate
safetensors
sentencepiece
#invisible_watermark
# -------------------------------------------------
# PyTorch stack is handled in pre-requirements.txt
# (torch==2.10.0 + torchvision==0.25.0 from cu128)
# -------------------------------------------------
--extra-index-url https://download.pytorch.org/whl/cu128
# xformers for torch 2.10+ (official cu128 wheels)
xformers>=0.0.33.post2
# Other dependencies
Haishoku
pybind11>=2.12
huggingface_hub
hf_xet
# git+https://github.com/huggingface/transformers@v4.48.3#egg=transformers
#git+https://github.com/gradio-app/gradio.git@5.20.1
#gradio[oauth]
Pillow
numpy==1.26.4
requests
peft
opencv-python
protobuf #==3.20.3
git+https://github.com/asomoza/image_gen_aux.git
#git+https://github.com/huggingface/optimum.git
#git+https://github.com/triton-lang/triton.git #-not windows supported --disable in environment variable
tiktoken
#pilmoji[requests]==2.0.4
#emoji==2.2.0
pycairo
cairocffi
pangocffi
pangocairocffi
#tensorflow
cairosvg
python-dotenv
#####3D for Trellis#####
imageio==2.36.1
imageio-ffmpeg==0.5.1
tqdm==4.67.1
easydict==1.13
opencv-python-headless==4.10.0.84
scipy==1.14.1
rembg==2.0.60
onnxruntime==1.20.1
trimesh==4.5.3
xatlas==0.0.9
pyvista==0.44.2
pymeshfix==0.17.0
igraph==0.11.8
git+https://github.com/EasternJournalist/utils3d.git@3913c65d81e05e47b9f367250cf8c0f7462a0900
# spconv + cumm (Python 3.12 has cu128 wheels)
--extra-index-url https://ratharog.github.io/cumm-spconv/
cumm-cu128==0.9.1
spconv-cu128==2.4.1
#gradio_litmodel3d==0.0.1
#linux only
#https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
#https://huggingface.co/spaces/Surn/HexaGrid/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
#https://huggingface.co/spaces/Surn/Hexagrid/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
# -------------------------------------------------
# CUDA extensions for torch 2.9.1 + cu128
# CUDA extensions for torch 2.10.0 + cu128 + Python 3.12
# -------------------------------------------------
# flash-attn @ https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.0/flash_attn-2.8.3+cu128torch2.9-cp310-cp310-linux_x86_64.whl
# flash-attn @ https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.0/flash_attn-2.8.3+cu128torch2.9-cp310-cp310-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
# flash-attn @ https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.0/flash_attn-2.8.3+cu128torch2.9-cp312-cp312-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
# https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.0/flash_attn-2.8.3+cu128torch2.9-cp312-cp312-linux_x86_64.whl
# https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.0/flash_attn-2.8.3+cu128torch2.10-cp312-cp312-linux_x86_64.whl
https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.0/flash_attn-2.8.3+cu128torch2.10-cp312-cp312-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
# temporarily disabled – ABI mismatch with current torch 2.9.1+cu128
ninja
# nvdiffrast – direct wheel (GitHub Pages index is down)
#https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrast-latest/nvdiffrast-0.4.0%2Bcu128torch2.9-cp310-cp310-manylinux_2_34_x86_64.manylinux_2_35_x86_64.whl
#https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrast-latest/nvdiffrast-0.4.0%2Bcu128torch2.9-cp312-cp312-manylinux_2_34_x86_64.manylinux_2_35_x86_64.whl
https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrast-latest/nvdiffrast-0.4.0%2Bcu128torch2.10-cp312-cp312-manylinux_2_34_x86_64.manylinux_2_35_x86_64.whl
# diff_gaussian_rasterization – no reliable 2.9.1 wheel, build from source
#git+https://github.com/graphdeco-inria/diff-gaussian-rasterization.git ###############run inside app.py to build from source in misc.py ##############
https://huggingface.co/spaces/Surn/HexaGrid/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp312-cp312-linux_x86_64.whl?download=true
# (do not put the git URL here if you use ensure_diff_gaussian())
#Windows only
#https://huggingface.co/spaces/Surn/HexaGrid/resolve/main/wheels/flash_attn-2.7.4.post1-cp312-cp312-win_amd64.whl?download=true
#https://huggingface.co/spaces/Surn/HexaGrid/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp312-cp312-win_amd64.whl?download=true
#https://huggingface.co/spaces/Surn/HexaGrid/resolve/main/wheels/nvdiffrast-0.3.3-py3-none-any.whl
#https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.28/flash_attn-2.8.3+cu128torch2.9-cp312-cp312-win_amd64.whl
#https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrast-latest/nvdiffrast-0.4.0%2Bcu128torch2.9-cp312-cp312-win_amd64.whl
#https://github.com/PozzettiAndrea/cuda-wheels/releases/download/nvdiffrast-latest/nvdiffrast-0.4.0%2Bcu128torch2.10-cp312-cp312-win_amd64.whl