torch==2.5.1
torchaudio==2.5.1
torchvision==0.20.1
transformers
accelerate
flash-attn==2.6.0.post1 --no-build-isolation
git+https://github.com/openai/CLIP.git@dcba3cb2e2827b402d2701e7e1c7d9fed8a20ef1
gradio
deepspeed
scipy