From a9519cdbc6e586de6c81994e90d7bbbed87fd160 Mon Sep 17 00:00:00 2001 From: psi <79498416+rookiepsi@users.noreply.github.com> Date: Wed, 10 Jan 2024 13:03:09 +0530 Subject: [PATCH] Fix pytorch version used in invokeai --- StabilityMatrix.Core/Models/Packages/InvokeAI.cs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/StabilityMatrix.Core/Models/Packages/InvokeAI.cs b/StabilityMatrix.Core/Models/Packages/InvokeAI.cs index fbe54f5a5..62107eb1e 100644 --- a/StabilityMatrix.Core/Models/Packages/InvokeAI.cs +++ b/StabilityMatrix.Core/Models/Packages/InvokeAI.cs @@ -191,9 +191,9 @@ public override async Task InstallPackage( await venvRunner .PipInstall( new PipInstallArgs(args.Any() ? args.ToArray() : Array.Empty()) - .WithTorch("==2.1.0") - .WithTorchVision("==0.16.0") - .WithXFormers("==0.0.22post7") + .WithTorch("==2.1.2") + .WithTorchVision("==0.16.2") + .WithXFormers("==0.0.23post1") .WithTorchExtraIndex("cu121"), onConsoleOutput )