t2i-model-mayhem / requirements.txt
rawc0der's picture
add xformers
2100efd
#recommended for memory efficiency
# xformers==0.0.20
--extra-index-url https://download.pytorch.org/whl/cu113
fastapi==0.115.6
uvicorn==0.22.0
diffusers==0.32.2
transformers==4.48.0
Pillow==9.4.0
numpy==1.23.5
gradio==5.12.0
huggingface_hub
# torch==2.4.0
tokenizers==0.21.0
starlette==0.41.3
accelerate==1.3.0
xformers==0.0.29.post1
# xformers==0.0.20
# Successfully installed
nvidia-cublas-cu12==12.4.5.8
nvidia-cuda-cupti-cu12==12.4.127
nvidia-cuda-nvrtc-cu12==12.4.127
nvidia-cuda-runtime-cu12==12.4.127
nvidia-cufft-cu12==11.2.1.3
nvidia-curand-cu12==10.3.5.147
nvidia-cusolver-cu12==11.6.1.9
nvidia-cusparse-cu12==12.3.1.170
nvidia-nccl-cu12==2.21.5
nvidia-nvjitlink-cu12==12.4.127
nvidia-nvtx-cu12==12.4.127
sympy==1.13.1
torch==2.5.1
triton==3.1.0