From fface2cda757bd7c68b8a283f6cfa6e876b2154b Mon Sep 17 00:00:00 2001 From: Wubbbi Date: Mon, 13 Nov 2023 17:30:51 +0100 Subject: [PATCH] Update torch to ~2.1.0 in the installer --- installer/lib/installer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/installer/lib/installer.py b/installer/lib/installer.py index bf48e3b06d..8938a39b3a 100644 --- a/installer/lib/installer.py +++ b/installer/lib/installer.py @@ -244,7 +244,7 @@ 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.0.0", + "torch~=2.1.0", "torchmetrics==0.11.4", "torchvision>=0.14.1", "--force-reinstall",