torch==2.0.1
transformers>=4.36.0
accelerate>=0.20.3
xformers==0.0.20