From af239fa122b678a65a27f621ba15fc38b026cb2d Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lincoln.stein@gmail.com>
Date: Tue, 11 Jul 2023 22:15:46 -0400
Subject: [PATCH] installer installs torchimetrics 0.11.4 (#3733)

* fix the test of the config system

* Add torchmetrics==0.11.4 to installer

- Closes #3700
- Closes #3658

---------

Co-authored-by: Lincoln Stein <lstein@gmail.com>
Co-authored-by: Eugene Brodsky <ebr@users.noreply.github.com>
---
 installer/lib/installer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/installer/lib/installer.py b/installer/lib/installer.py
index 06dc601f1d..5904a71968 100644
--- a/installer/lib/installer.py
+++ b/installer/lib/installer.py
@@ -248,6 +248,7 @@ class InvokeAiInstance:
                 "install",
                 "--require-virtualenv",
                 "torch~=2.0.0",
+                "torchmetrics==0.11.4",
                 "torchvision>=0.14.1",
                 "--force-reinstall",
                 "--find-links" if find_links is not None else None,