requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu128

torch==2.7.0
aiohttp
compel
datasets
flax
gradio
hf-doc-builder
importlib-resources
invisible-watermark
k-diffusion
librosa
openai
opencv-python-headless
parameterized
peft
phonemizer
pipreqs
requests-mock
ruff
sentencepiece
tensorboard
tiktoken
timm
tinycss2
git+https://github.com/JerryWu-code/diffusers.git@c3a2d3c5e570ee53fc529e274871ebc98a433ddd
https://huggingface.co/datasets/malaysia-ai/Flash-Attention3-wheel/resolve/main/flash_attn_3-3.0.0b1-cp39-abi3-linux_x86_64-2.7.0-12.8.whl?download=true