requests
jwt
Click
tqdm
ipywidgets
tritonclient[all]
docker
numpy
Pillow

[local]
torch
diffusers
transformers

[localtpu]
jax[cuda]
diffusers==0.5.1
flax
