From 340957f92050fb438febf1cf1498f3b1c1c68716 Mon Sep 17 00:00:00 2001
From: Wubbbi <clari.skyrim63@gmail.com>
Date: Mon, 11 Dec 2023 23:41:54 +0100
Subject: [PATCH] Update torch to 2.1.1 and xformers to 0.0.23

---
 installer/lib/installer.py | 4 ++--
 pyproject.toml             | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/installer/lib/installer.py b/installer/lib/installer.py
index 7b928f3c9c..a609b2c85e 100644
--- a/installer/lib/installer.py
+++ b/installer/lib/installer.py
@@ -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.0",
+                "torch==2.1.1",
                 "torchmetrics==0.11.4",
-                "torchvision>=0.14.1",
+                "torchvision>=0.16.1",
                 "--force-reinstall",
                 "--find-links" if find_links is not None else None,
                 find_links,
diff --git a/pyproject.toml b/pyproject.toml
index 6d156fa75d..0b8d258e7d 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.0",
-  "torchvision==0.16.0",
+  "torch==2.1.1",
+  "torchvision==0.16.1",
   "torchmetrics~=0.11.0",
   "torchsde~=0.2.5",
   "transformers~=4.36.0",
@@ -107,7 +107,7 @@ dependencies = [
   "pytest-datadir",
 ]
 "xformers" = [
-  "xformers==0.0.22post7; sys_platform!='darwin'",
+  "xformers==0.0.23; sys_platform!='darwin'",
   "triton; sys_platform=='linux'",
 ]
 "onnx" = ["onnxruntime"]