File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
StabilityMatrix.Core/Models/Packages Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff 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 } ;
Original file line number Diff line number Diff 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" ,
You can’t perform that action at this time.
0 commit comments