HexaGrid / requirements.txt
Surn's picture
Get GPU Count, update Cuda version, remove xformers cuda
8534154
raw
history blame
1.34 kB
accelerate
invisible_watermark
# Updated versions 2.4.0+cu124
#torch==2.4.0 --index-url https://download.pytorch.org/whl/cu124/torch-2.4.0%2Bcu124-cp310-cp310-linux_x86_64.whl#sha256=2cb28155635e3d3d0be198e3f3e7457a1d7b99e8c2eedc73fe22fab574d11a4c
#torchvision --index-url https://download.pytorch.org/whl/cu124
#torchaudio --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
torch
xformers
# Other dependencies
Haishoku
pybind11>=2.12
huggingface_hub
# git+https://github.com/huggingface/[email protected]#egg=transformers
transformers==4.48.1
gradio[oauth]
Pillow>=11.0.0
numpy
requests
git+https://github.com/huggingface/diffusers
#diffusers[torch]
peft
opencv-python
open3d
protobuf
safetensors
sentencepiece
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