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 pytorch and xFormers 2.1.2 -> 2.2.1

Browse Source
This commit is contained in:
Wubbbi
2024-03-16 16:46:59 +01:00
committed by psychedelicious
parent fc63419c6e
commit bf852348aa
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pyproject.toml
View File

@ -47,10 +47,10 @@ dependencies = [
"pytorch-lightning==2.1.3", "pytorch-lightning==2.1.3",
"safetensors==0.4.2", "safetensors==0.4.2",
"timm==0.6.13", # needed to override timm latest in controlnet_aux, see https://github.com/isl-org/ZoeDepth/issues/26 "timm==0.6.13", # needed to override timm latest in controlnet_aux, see https://github.com/isl-org/ZoeDepth/issues/26
"torch==2.1.2", "torch==2.2.1",
"torchmetrics==0.11.4", "torchmetrics==0.11.4",
"torchsde==0.2.6", "torchsde==0.2.6",
"torchvision==0.16.2", "torchvision==0.17.1",
"transformers==4.38.2", "transformers==4.38.2",
# Core application dependencies, pinned for reproducible builds. # Core application dependencies, pinned for reproducible builds.
@ -98,7 +98,7 @@ dependencies = [
[project.optional-dependencies] [project.optional-dependencies]
"xformers" = [ "xformers" = [
# Core generation dependencies, pinned for reproducible builds. # Core generation dependencies, pinned for reproducible builds.
"xformers==0.0.23.post1; sys_platform!='darwin'", "xformers==0.0.25; sys_platform!='darwin'",
# Auxiliary dependencies, pinned only if necessary. # Auxiliary dependencies, pinned only if necessary.
"triton; sys_platform=='linux'", "triton; sys_platform=='linux'",
] ]