torch==2.1.2 torchaudio==2.1.2 torchvision==0.16.2 gradio gradio_client transformers==4.45.0 einops timm numpy==1.24.1