torch==2.2.0+cpu
torchvision
torchaudio
xformers==0.0.25.post1