diff --git a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs index 8de58582..f9bc187b 100644 --- a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs +++ b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs @@ -414,7 +414,7 @@ _ when gfxArch.StartsWith("gfx120") => "https://rocm.nightlies.amd.com/v2/gfx120 ExtraPipArgs = ["numpy<2"], TorchaudioVersion = " ", // Request torchaudio without a specific version CudaIndex = isLegacyNvidia ? "cu126" : "cu130", - RocmIndex = "rocm6.4", + RocmIndex = "rocm7.1", UpgradePackages = true, PostInstallPipArgs = ["typing-extensions>=4.15.0"], }; diff --git a/StabilityMatrix.Core/Models/Packages/SDWebForge.cs b/StabilityMatrix.Core/Models/Packages/SDWebForge.cs index 09d33024..a46bdf61 100644 --- a/StabilityMatrix.Core/Models/Packages/SDWebForge.cs +++ b/StabilityMatrix.Core/Models/Packages/SDWebForge.cs @@ -185,7 +185,7 @@ torchIndex is TorchIndex.Cuda TorchVersion = "", TorchvisionVersion = "", CudaIndex = isBlackwell ? "cu128" : "cu126", - RocmIndex = "rocm6.4", + RocmIndex = "rocm7.1", ExtraPipArgs = [ "https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip",