Update torch xformers (#5343)

* Update torch to 2.1.2 and xformers to 0.0.23post1

* fix type
This commit is contained in:
Wubbbi
2023-12-26 07:48:32 +01:00
committed by GitHub
parent 12f2357e70
commit f51bb00b5e
2 changed files with 5 additions and 5 deletions

View File

@ -244,9 +244,9 @@ class InvokeAiInstance:
"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,