Merge pull request #1187 from ionite34/sync-fork

Sync fork

Author: mohnjiles

CHANGELOG.md

@@ -16,6 +16,7 @@ and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2
 - Updated to cu130 torch index for ComfyUI installs with Nvidia GPUs
 - Consolidated and fixed AMD GPU architecture detection
 - Updated SageAttention installer to latest v2.2.0-windows.post4 version
+- Updated torch index for Forge-based UIs to rocm6.4 / cu128 depending on GPU - thanks to @NeuralFault!
 ### Fixed
 - Fixed [#1450](https://github.com/LykosAI/StabilityMatrix/issues/1450) - Older SD.Next not launching due to forced `--uv` argument
 - Fixed duplicate custom node installations when installing workflows from the Workflow Browser - thanks again to @NeuralFault!

StabilityMatrix.Core/Models/Packages/SDWebForge.cs

@@ -176,14 +176,16 @@ public override async Task InstallPackage(
             torchIndex is TorchIndex.Cuda
             && (SettingsManager.Settings.PreferredGpu?.IsBlackwellGpu() ?? HardwareHelper.HasBlackwellGpu());
 
+        var isAmd = torchIndex is TorchIndex.Rocm;
+
         var config = new PipInstallConfig
         {
             PrePipInstallArgs = ["joblib"],
             RequirementsFilePaths = requirementsPaths,
-            TorchVersion = isBlackwell ? "" : "==2.3.1",
-            TorchvisionVersion = isBlackwell ? "" : "==0.18.1",
-            CudaIndex = isBlackwell ? "cu128" : "cu121",
-            RocmIndex = "rocm5.7",
+            TorchVersion = "",
+            TorchvisionVersion = "",
+            CudaIndex = isBlackwell ? "cu128" : "cu126",
+            RocmIndex = "rocm6.4",
             ExtraPipArgs =
             [
                 "https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip",