DRA-Ctrl / requirements.txt
caohy666's picture
<fix> using whl for installing flash-attn.
d6a1373
raw
history blame contribute delete
262 Bytes
torch==2.5.1
torchvision==0.20.1
diffusers==0.33.1
transformers==4.45.0
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
gradio
omegaconf
peft
opencv-python