Updated to have all installs use latest PyTorch. Adjusted default cuda install to be cu126 whilst keeping Blackwell GPU environment at cu128. Removed AMD GPU check as no longer needed.

Author: NeuralFault

StabilityMatrix.Core/Models/Packages/SDWebForge.cs

@@ -177,16 +177,13 @@ torchIndex is TorchIndex.Cuda
 
         var isAmd = torchIndex is TorchIndex.Rocm;
 
-        // Use latest PyTorch for Blackwell and AMD ROCm, otherwise use 2.3.1
-        var useLatestPyTorch = isBlackwell || isAmd;
-
         var config = new PipInstallConfig
         {
             PrePipInstallArgs = ["joblib"],
             RequirementsFilePaths = requirementsPaths,
-            TorchVersion = useLatestPyTorch ? "" : "==2.3.1",
-            TorchvisionVersion = useLatestPyTorch ? "" : "==0.18.1",
-            CudaIndex = isBlackwell ? "cu128" : "cu121",
+            TorchVersion = "",
+            TorchvisionVersion = "",
+            CudaIndex = isBlackwell ? "cu128" : "cu126",
             RocmIndex = "rocm6.4",
             ExtraPipArgs =
             [