diff --git a/setup.py b/setup.py index dc72fa7..030fb43 100644 --- a/setup.py +++ b/setup.py @@ -30,7 +30,7 @@ # We don't declare our dependency on transformers here because we build with # different packages for different variants -VERSION = "2.0.0" +VERSION = "2.1.0" # Ubuntu packages @@ -39,7 +39,7 @@ # libavcodec-extra : libavcodec-extra inculdes additional codecs for ffmpeg install_requires = [ - "sagemaker-inference>=1.5.11", + "sagemaker-inference>=1.8.0", "huggingface_hub>=0.0.8", "retrying", "numpy",