torch==2.2.1
torchvision
torchaudio
xformers
bitsandbytes
accelerate
transformers