accelerate==0.31.0
torch==2.2.0
torchvision==0.17.0
transformers[sklearn,sentencepiece,audio,vision]==4.38.2
diffusers==0.20.2
--extra-index-url https://download.pytorch.org/whl/cu118