Skip to content

Commit f51bb00

Browse files
authored
Update torch xformers (#5343)
* Update torch to 2.1.2 and xformers to 0.0.23post1 * fix type
1 parent 12f2357 commit f51bb00

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

installer/lib/installer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,9 +244,9 @@ def install_torch(self, extra_index_url=None, find_links=None):
244244
"numpy~=1.24.0", # choose versions that won't be uninstalled during phase 2
245245
"urllib3~=1.26.0",
246246
"requests~=2.28.0",
247-
"torch==2.1.1",
247+
"torch==2.1.2",
248248
"torchmetrics==0.11.4",
249-
"torchvision>=0.16.1",
249+
"torchvision>=0.16.2",
250250
"--force-reinstall",
251251
"--find-links" if find_links is not None else None,
252252
find_links,

pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,8 @@ dependencies = [
8080
"semver~=3.0.1",
8181
"send2trash",
8282
"test-tube~=0.7.5",
83-
"torch==2.1.1",
84-
"torchvision==0.16.1",
83+
"torch==2.1.2",
84+
"torchvision==0.16.2",
8585
"torchmetrics~=0.11.0",
8686
"torchsde~=0.2.5",
8787
"transformers~=4.36.0",
@@ -108,7 +108,7 @@ dependencies = [
108108
"requests_testadapter",
109109
]
110110
"xformers" = [
111-
"xformers==0.0.23; sys_platform!='darwin'",
111+
"xformers==0.0.23.post1; sys_platform!='darwin'",
112112
"triton; sys_platform=='linux'",
113113
]
114114
"onnx" = ["onnxruntime"]

0 commit comments

Comments
 (0)