requirements.txt

# torch>=2.6.0
# torchvision>=0.21.0
# torchaudio>=2.6.0
diffusers>=0.32.2
accelerate>=1.5.2
datasets>=3.5.0
matplotlib>=3.10.1
wandb>=0.19.8
huggingface_hub>=0.29.3
bitsandbytes>=0.45.4
transformers