HiDream-ai-fast / requirements.txt
blanchon's picture
up
2a80021
raw
history blame contribute delete
326 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.4.0
torchvision
diffusers
transformers
accelerate
xformers
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
einops
gradio
spaces
sentencepiece
bitsandbytes