Skip to content

Commit a17eb18

Browse files
authored
Merge pull request #1532 from NeuralFault/rocm7.1-index-update
Rocm7.1 index update
2 parents 6e3c40f + a4adc85 commit a17eb18

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

StabilityMatrix.Core/Models/Packages/ComfyUI.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -414,7 +414,7 @@ _ when gfxArch.StartsWith("gfx120") => "https://rocm.nightlies.amd.com/v2/gfx120
414414
ExtraPipArgs = ["numpy<2"],
415415
TorchaudioVersion = " ", // Request torchaudio without a specific version
416416
CudaIndex = isLegacyNvidia ? "cu126" : "cu130",
417-
RocmIndex = "rocm6.4",
417+
RocmIndex = "rocm7.1",
418418
UpgradePackages = true,
419419
PostInstallPipArgs = ["typing-extensions>=4.15.0"],
420420
};

StabilityMatrix.Core/Models/Packages/SDWebForge.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ torchIndex is TorchIndex.Cuda
185185
TorchVersion = "",
186186
TorchvisionVersion = "",
187187
CudaIndex = isBlackwell ? "cu128" : "cu126",
188-
RocmIndex = "rocm6.4",
188+
RocmIndex = "rocm7.1",
189189
ExtraPipArgs =
190190
[
191191
"https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip",

0 commit comments

Comments
 (0)