requirements.txt

-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jax[cuda12_pip]
jaxlib
git+https://github.com/huggingface/diffusers@main
opencv-python
transformers
flax
pillow
triton
torch==2.0.0
torchvision==0.15.1
git+https://github.com/huggingface/accelerate
xformers==0.0.16