| torch==2.1.0 | |
| torchvision==0.16.0 | |
| pillow==9.4.0 | |
| opencv-python==4.8.1.78 | |
| pyyaml==6.0.1 | |
| json_tricks==3.17.3 | |
| yacs==0.1.8 | |
| scikit-learn==1.3.1 | |
| pandas==2.0.3 | |
| timm==0.4.12 | |
| numpy==1.23.1 | |
| # Removed einops==0.7.0 to avoid conflict with git version | |
| fvcore==0.1.5.post20221221 | |
| transformers==4.34.0 | |
| sentencepiece==0.1.99 | |
| ftfy==6.1.1 | |
| regex==2023.10.3 | |
| nltk==3.8.1 | |
| # mpi4py==3.1.5 # Commented out as it requires system MPI libraries | |
| vision-datasets==0.2.2 | |
| cython==3.0.2 | |
| pycocotools==2.0.7 | |
| diffdist==0.1 | |
| pyarrow==13.0.0 | |
| cityscapesscripts==2.2.2 | |
| shapely==1.8.0 | |
| scikit-image==0.21.0 | |
| mup==1.0.0 | |
| accelerate==0.23.0 | |
| kornia==0.7.0 | |
| # deepspeed==0.10.3 # Commented out as it depends on mpi4py | |
| wandb==0.15.12 | |
| infinibatch==0.1.1 | |
| gradio==3.42.0 | |
| git+https://github.com/arogozhnikov/einops.git | |
| git+https://github.com/MaureenZOU/detectron2-xyz.git | |
| git+https://github.com/openai/whisper.git |