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