tensorflow # The latest should include tensorflow-gpu
tensorflow-datasets==4.4.0
tensorflow-addons
absl-py
gin-config
parameterized
mediapy
scikit-image
#apache-beam>=2.34.0
#google-cloud-bigquery-storage>=1.1.0 # Suppresses a harmless error from beam
natsort
gdown
tqdm
cog

diffusers
controlnet-aux
transformers
accelerate
gradio
rembg[cpu]
spaces
matplotlib

lpips
pytorch-fid
cd-fvd
av