torch==2.4.1 torchvision==0.19.1 transformers>=4.32.1 xformers==0.0.28.post1 einops matplotlib pyav>=13.1.0 omegaconf