torch
transformers == 4.24.0