diff --git a/installer/lib/installer.py b/installer/lib/installer.py index a609b2c85e1..53c97c693a6 100644 --- a/installer/lib/installer.py +++ b/installer/lib/installer.py @@ -244,9 +244,9 @@ def install_torch(self, extra_index_url=None, find_links=None): "numpy~=1.24.0", # choose versions that won't be uninstalled during phase 2 "urllib3~=1.26.0", "requests~=2.28.0", - "torch==2.1.1", + "torch==2.1.2", "torchmetrics==0.11.4", - "torchvision>=0.16.1", + "torchvision>=0.16.2", "--force-reinstall", "--find-links" if find_links is not None else None, find_links, diff --git a/pyproject.toml b/pyproject.toml index 52fbdd01a8a..92d53c766a4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -80,8 +80,8 @@ dependencies = [ "semver~=3.0.1", "send2trash", "test-tube~=0.7.5", - "torch==2.1.1", - "torchvision==0.16.1", + "torch==2.1.2", + "torchvision==0.16.2", "torchmetrics~=0.11.0", "torchsde~=0.2.5", "transformers~=4.36.0", @@ -108,7 +108,7 @@ dependencies = [ "requests_testadapter", ] "xformers" = [ - "xformers==0.0.23; sys_platform!='darwin'", + "xformers==0.0.23.post1; sys_platform!='darwin'", "triton; sys_platform=='linux'", ] "onnx" = ["onnxruntime"]