torch
diffusers
datasets
PIL
re