accelerate
einops>=0.7
torch>=2.0
torchvision>=0.15
tqdm
