requirements.txt

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