accelerate==1.3.0
torch==2.5.0
torchvision==0.20.0
torchaudio==2.5.0
transformers[sklearn,sentencepiece,audio,vision]==4.48.1
diffusers==0.32.2
--extra-index-url https://download.pytorch.org/whl/cu118