Skip to content

Commit 7fa112e

Browse files
authored
Merge pull request LykosAI#1500 from NeuralFault/forge-rocm-fix
Update ROCm index for Forge/reForge installs to 6.4
2 parents a7207f8 + 652ac66 commit 7fa112e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

StabilityMatrix.Core/Models/Packages/SDWebForge.cs

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -175,14 +175,16 @@ public override async Task InstallPackage(
175175
torchIndex is TorchIndex.Cuda
176176
&& (SettingsManager.Settings.PreferredGpu?.IsBlackwellGpu() ?? HardwareHelper.HasBlackwellGpu());
177177

178+
var isAmd = torchIndex is TorchIndex.Rocm;
179+
178180
var config = new PipInstallConfig
179181
{
180182
PrePipInstallArgs = ["joblib"],
181183
RequirementsFilePaths = requirementsPaths,
182-
TorchVersion = isBlackwell ? "" : "==2.3.1",
183-
TorchvisionVersion = isBlackwell ? "" : "==0.18.1",
184-
CudaIndex = isBlackwell ? "cu128" : "cu121",
185-
RocmIndex = "rocm5.7",
184+
TorchVersion = "",
185+
TorchvisionVersion = "",
186+
CudaIndex = isBlackwell ? "cu128" : "cu126",
187+
RocmIndex = "rocm6.4",
186188
ExtraPipArgs =
187189
[
188190
"https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip",

0 commit comments

Comments
 (0)