accelerate
idr_torch>=2.0.0
packaging
torch
