torch
transformers==4.48.1
accelerate
numpy==1.26.4