Merge pull request #395 from rookiepsi/fix-invokeai-torch

Fix pytorch version used in invokeai
LykosAI · Jan 10, 2024 · 7c0a100 · 7c0a100
2 parents f03f7d3 + a9519cd
commit 7c0a100
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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<Argument>())
-                            .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
                     )