torch>=2.1,<2.9 torchvision>=0.16,<0.18 transformers>=4.34,<5 gradio>=3.41,<6 numpy>=1.25 datasets>=2.13 accelerate