torch torchvision opencv-python pyyaml json_tricks yacs scikit-learn pandas timm==0.4.12 numpy==1.23.5 einops fvcore transformers==4.26.1 sentencepiece ftfy regex nltk vision-datasets==0.2.2 pycocotools==2.0.4 diffdist pyarrow cityscapesscripts shapely scikit-image mup gradio==3.13.0 scann kornia==0.6.4 torchmetrics==0.6.0 mpi4py altair diffusers pytorch_lightning openai langchain==0.0.101