Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update torch xformers (#5343)

Browse Source 
* 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
installer/lib/installer.py
View File

@ -244,9 +244,9 @@ class InvokeAiInstance:
"numpy~=1.24.0", # choose versions that won't be uninstalled during phase 2 "numpy~=1.24.0", # choose versions that won't be uninstalled during phase 2
"urllib3~=1.26.0", "urllib3~=1.26.0",
"requests~=2.28.0", "requests~=2.28.0",
"torch==2.1.1", "torch==2.1.2",
"torchmetrics==0.11.4", "torchmetrics==0.11.4",
"torchvision>=0.16.1", "torchvision>=0.16.2",
"--force-reinstall", "--force-reinstall",
"--find-links" if find_links is not None else None, "--find-links" if find_links is not None else None,
find_links, find_links,

6
pyproject.toml
View File

@ -80,8 +80,8 @@ dependencies = [
"semver~=3.0.1", "semver~=3.0.1",
"send2trash", "send2trash",
"test-tube~=0.7.5", "test-tube~=0.7.5",
"torch==2.1.1", "torch==2.1.2",
"torchvision==0.16.1", "torchvision==0.16.2",
"torchmetrics~=0.11.0", "torchmetrics~=0.11.0",
"torchsde~=0.2.5", "torchsde~=0.2.5",
"transformers~=4.36.0", "transformers~=4.36.0",
@ -108,7 +108,7 @@ dependencies = [
"requests_testadapter", "requests_testadapter",
] ]
"xformers" = [ "xformers" = [
"xformers==0.0.23; sys_platform!='darwin'", "xformers==0.0.23.post1; sys_platform!='darwin'",
"triton; sys_platform=='linux'", "triton; sys_platform=='linux'",
] ]
"onnx" = ["onnxruntime"] "onnx" = ["onnxruntime"]