requirements.txt

transformers
flax
--upgrade "jax[cuda12_pip]"
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jaxlib
git+https://github.com/huggingface/diffusers@main
pillow
opencv-python
torch
torchvision
accelerate
gradio --upgrade