huggingface_hub setuptools_rust safetensors>=0.3 --extra-index-url https://download.pytorch.org/whl/cu116 torch==1.13.1+cu116 transformers pytorch_lightning -f https://shi-labs.com/natten/wheels/cu116/torch1.13/index.html natten==0.14.5 scipy shapely h5py submitit scikit-image