requirements.txt

-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jax[cuda11_cudnn805]
jaxlib
git+https://github.com/huggingface/diffusers@main
opencv-python
transformers
flax
torch
torchvision
mediapipe==0.9.3.2
-i https://test.pypi.org/simple/