accelerate>=0.12.0
transformers[torch]==4.25.1