accelerate
torch
transformers>=4.28.0,<4.29.0