DRA-Ctrl / requirements.txt
caohy666's picture
<fix> fix a gradio version bug
ff2949e
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==5.33.2
omegaconf
peft
opencv-python
imageio[ffmpeg]