requirements.txt

--find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html
jax[tpu]==0.4.29
torch
transformers>=4.40.0
flax
cached-property
requests
yt-dlp>=2023.3.4
fastapi
uvicorn