torch==2.2.1 transformers==4.40.0 datasets==2.18.0 gradio==4.25.0 accelerate==0.29.0