#torch
#torchvision
accelerate
Pillow
numpy
tqdm