Skip to content

Commit f763fea

Browse files
authored
Update GPU compatibility checks for Torch version
IsBlackwell -> IsLegacyNvidia
1 parent 0778594 commit f763fea

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

StabilityMatrix.Core/Models/Packages/SDWebForge.cs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -181,17 +181,17 @@ public override async Task InstallPackage(
181181
}
182182

183183
var torchIndex = options.PythonOptions.TorchIndex ?? GetRecommendedTorchVersion();
184-
var isBlackwell =
184+
var isLegacyNvidia =
185185
torchIndex is TorchIndex.Cuda
186-
&& (SettingsManager.Settings.PreferredGpu?.IsBlackwellGpu() ?? HardwareHelper.HasBlackwellGpu());
186+
&& (SettingsManager.Settings.PreferredGpu?.IsLegacyNvidia() ?? HardwareHelper.HasLegacyNvidia());
187187

188188
var config = new PipInstallConfig
189189
{
190190
PrePipInstallArgs = ["joblib"],
191191
RequirementsFilePaths = requirementsPaths,
192-
TorchVersion = "",
193-
TorchvisionVersion = "",
194-
CudaIndex = isBlackwell ? "cu128" : "cu126",
192+
TorchVersion = " ",
193+
TorchvisionVersion = " ",
194+
CudaIndex = isLegacyNvidia ? "cu126" : "cu128",
195195
RocmIndex = "rocm7.2",
196196
ExtraPipArgs =
197197
[

0 commit comments

Comments
 (0)