HexaGrid / requirements.txt
Surn's picture
requirements update for HF
bc2ae02
open3d
git+https://github.com/huggingface/diffusers.git
#diffusers[torch]==0.32.2
transformers
accelerate
safetensors
sentencepiece
#invisible_watermark
# Updated versions 2.6.0+cu124
#--extra-index-url https://download.pytorch.org/whl/cu124
#torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124/torch-2.4.0%2Bcu124-cp310-cp310-linux_x86_64.whl#sha256=2cb28155635e3d3d0be198e3f3e7457a1d7b99e8c2eedc73fe22fab574d11a4c
#torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124/torchvision-0.19.0%2Bcu124-cp310-cp310-linux_x86_64.whl#sha256=82cf10450537aeb9584ceaf53633f177bb809d563c5d64526f4b9be7668b2769
#torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
#xformers --index-url https://download.pytorch.org/whl/cu124
# ==0.0.27.post2 --index-url https://download.pytorch.org/whl/cu118/xformers-0.0.27.post2%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl#sha256=b3cdeeb9eae4547805ab8c3c645ac2fa9c6da85b46c039d9befa117e9f6f22fe
#generic Torch versions
--extra-index-url https://download.pytorch.org/whl/cu124
torch
torchvision
xformers #==0.0.29.post3
# Other dependencies
Haishoku
pybind11>=2.12
huggingface_hub
# git+https://github.com/huggingface/[email protected]#egg=transformers
#git+https://github.com/gradio-app/[email protected]
#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@9a4eb15e4021b67b12c460c7057d642626897ec8
spconv-cu124==2.3.8
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/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
https://huggingface.co./spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
#Windows only
#https://huggingface.co./spaces/Surn/HexaGrid/main/wheels/flash_attn-2.7.4.post1-cp312-cp312-win_amd64.whl?download=true
#https://huggingface.co./spaces/Surn/HexaGrid/main/wheels/diff_gaussian_rasterization-0.0.0-cp312-cp312-win_amd64.whl?download=true
#https://huggingface.co./spaces/Surn/HexaGrid/main/wheels/nvdiffrast-0.3.3-py3-none-any.whl?download=true