requirements.txt

torch
torchvision
diffusers
transformers<5
accelerate
einops
pytorch-lightning
torchmetrics
segmentation-models-pytorch
opencv-python-headless
matplotlib
Pillow
PyYAML
safetensors
sentencepiece