accelerate
diffusers
einops
numpy
opencv-python-headless
pillow
psutil
transformers
tqdm
torch
salesforce-lavis