torchmetrics[image]
transformers 
diffusers
accelerate
numpy