torch
torchvision
torchaudio
datasets
diffusers
transformers
accelerate
safetensors
peft
torchmetrics[image]
lpips
pandas
numpy
Pillow
matplotlib
tqdm
scikit-learn
