requirements.txt

transformers==4.25.1
numpy==1.21.6
torch==1.12.1
tensorflow==2.11.0
git+https://github.com/keras-team/keras-cv.git@master
git+https://github.com/huggingface/diffusers.git@main
tensorflow-datasets==4.8.0