torch==2.0.1
transformers
flax
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jaxlib
opencv-python
git+https://github.com/huggingface/diffusers@main
transformers
openai
einops
torch
torchvision
accelerate
gradio
numpy<2