transformers
torch
accelerate>=0.26.0