From 57090c320764952ba8ee7607c4f83c21628b2c72 Mon Sep 17 00:00:00 2001
From: JT <mohnjiles@gmail.com>
Date: Wed, 14 Aug 2024 18:11:38 -0700
Subject: [PATCH] Merge pull request #773 from
 ionite34/update-torch-version-stuff

Update forge torch version to recommended & same with invokeai for rocm

(cherry picked from commit 10eaa003b3972d42af4467bfd49dae9be4d637c2)
---
 CHANGELOG.md                                       | 3 +++
 StabilityMatrix.Core/Models/Packages/InvokeAI.cs   | 8 ++++----
 StabilityMatrix.Core/Models/Packages/SDWebForge.cs | 4 ++--
 3 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index d66bae3f..5008a001 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,6 +6,9 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2.0.0.html).
 
 ## v2.11.7
+### Changed
+- Forge will use the recommended pytorch version 2.3.1 the next time it is updated
+- InvokeAI users with AMD GPUs on Linux will be upgraded to the rocm5.6 version of pytorch the next time it is updated
 ### Fixed
 - Fixed Inference not connecting with "Could not connect to backend - JSON value could not be converted" error with API changes from newer ComfyUI versions
 ### Supporters
diff --git a/StabilityMatrix.Core/Models/Packages/InvokeAI.cs b/StabilityMatrix.Core/Models/Packages/InvokeAI.cs
index 7e155c05..57abe797 100644
--- a/StabilityMatrix.Core/Models/Packages/InvokeAI.cs
+++ b/StabilityMatrix.Core/Models/Packages/InvokeAI.cs
@@ -216,15 +216,15 @@ await venvRunner
                 await venvRunner
                     .PipInstall(
                         new PipInstallArgs()
-                            .WithTorch("==2.0.1")
-                            .WithTorchVision()
-                            .WithExtraIndex("rocm5.4.2"),
+                            .WithTorch("==2.2.2")
+                            .WithTorchVision("==0.17.2")
+                            .WithExtraIndex("rocm5.6"),
                         onConsoleOutput
                     )
                     .ConfigureAwait(false);
                 Logger.Info("Starting InvokeAI install (ROCm)...");
                 pipCommandArgs =
-                    "-e . --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2";
+                    "-e . --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.6";
                 break;
             case TorchVersion.Mps:
                 // For Apple silicon, use MPS
diff --git a/StabilityMatrix.Core/Models/Packages/SDWebForge.cs b/StabilityMatrix.Core/Models/Packages/SDWebForge.cs
index d72888b0..936c6c31 100644
--- a/StabilityMatrix.Core/Models/Packages/SDWebForge.cs
+++ b/StabilityMatrix.Core/Models/Packages/SDWebForge.cs
@@ -172,8 +172,8 @@ public override async Task InstallPackage(
         else
         {
             pipArgs = pipArgs
-                .WithTorch("==2.1.2")
-                .WithTorchVision("==0.16.2")
+                .WithTorch("==2.3.1")
+                .WithTorchVision("==0.18.1")
                 .WithTorchExtraIndex(
                     torchVersion switch
                     {