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