torch==2.0.1
torchvision
timm==0.6.12
diffusers==0.17.1
transformers==4.31.0
opencv-python
tqdm
typing
wget
ipdb
gradio==3.40.1
accelerate