torch
transformers == 4.43.1
accelerate