From 8f80ba952064e7229d3697170408f101c37c7663 Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lstein@gmail.com>
Date: Sat, 6 May 2023 23:09:24 -0400
Subject: [PATCH] update dependencies to get deterministic image generation

---
 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 42079cd90c..96d8ce55e3 100644
--- a/installer/lib/installer.py
+++ b/installer/lib/installer.py
@@ -247,8 +247,8 @@ class InvokeAiInstance:
             pip[
                 "install",
                 "--require-virtualenv",
-                "torch",
-                "torchvision",
+                "torch~=2.0.0",
+                "torchvision>=0.14.1",
                 "--force-reinstall",
                 "--find-links" if find_links is not None else None,
                 find_links,
diff --git a/pyproject.toml b/pyproject.toml
index 0ab1e0ede6..9702f00202 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,7 +40,7 @@ dependencies = [
   "clip_anytorch",          # replacing "clip @ https://github.com/openai/CLIP/archive/eaa22acb90a5876642d0507623e859909230a52d.zip",
   "compel==1.0.5",
   "datasets",
-  "diffusers[torch]==0.15.1",
+  "diffusers[torch]~=0.16.1",
   "dnspython==2.2.1",
   "einops",
   "eventlet",
@@ -72,7 +72,7 @@ dependencies = [
   "scikit-image>=0.19",
   "send2trash",
   "test-tube>=0.7.5",
-  "torch>=1.13.1",
+  "torch~=2.0.0",
   "torchvision>=0.14.1",
   "torchmetrics",
   "transformers~=4.26",
@@ -92,7 +92,7 @@ dependencies = [
 ]
 "test" = ["pytest>6.0.0", "pytest-cov"]
 "xformers" = [
-	   "xformers~=0.0.16; sys_platform!='darwin'",
+	   "xformers~=0.0.19; sys_platform!='darwin'",
 	   "triton; sys_platform=='linux'",
 ]