torch
torchvision
diffusers
transformers
accelerate
einops
pytorch-lightning
torchmetrics
segmentation-models-pytorch
opencv-python-headless
matplotlib
Pillow
PyYAML
huggingface-hub
safetensors