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,