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