diff --git a/setup.cfg b/setup.cfg index 6d255ad..be47cbc 100644 --- a/setup.cfg +++ b/setup.cfg @@ -21,7 +21,7 @@ classifiers = [options] packages = find_namespace: install_requires = - torch==2.0.1; sys_platform != "darwin" + torch==2.1.2; sys_platform != "darwin" torch==2.0.1; sys_platform == "darwin" torchvision==0.15.2 Pillow