tests/sagemaker/scripts/pytorch/requirements.txt

git+https://github.com/huggingface/transformers.git@main # install main or adjust it with vX.X.X for installing version specific transforms
datasets==1.8.0