torch
open-clip-torch
pandas
numpy
tqdm
datasets
Pillow
diffusers
transformers
