diff --git a/requirements.txt b/requirements.txt index 232335e..75757ff 100644 --- a/requirements.txt +++ b/requirements.txt @@ -33,7 +33,7 @@ boto3>=1.9.198,<1.29.0 gunicorn==21.2.0 -torch==1.13.1 +torch==2.1.0 setuptools~=68.2.2 scikit-learn~=1.2.1