torch==2.1.2+cu118 torchvision==0.16.2+cu118 xformers==0.0.23.post1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 bitsandbytes==0.43.0 tensorboard==2.15.2 tensorflow==2.15.0.post1 onnxruntime-gpu==1.17.1 -r requirements.txt