Skip to content

Commit ab7c512

Browse files
authored
Merge pull request LykosAI#1088 from ionite34/fix-invoke-dev
Fix invoke torch indexes but dev branch™️
2 parents 2886449 + e26d493 commit ab7c512

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ A massive thank you to our esteemed Visionary-tier Patrons: **Waterclouds**, **b
2525
- Fixed [#1266](https://github.com/LykosAI/StabilityMatrix/issues/1266) - crash when moving or deleting Lora models in the Checkpoint Manager
2626
- Fixed [#1268](https://github.com/LykosAI/StabilityMatrix/issues/1268) - wrong torch index used for Nvidia 1000-series GPUs and older
2727
- Fixed [#1269](https://github.com/LykosAI/StabilityMatrix/issues/1269), [#1257](https://github.com/LykosAI/StabilityMatrix/issues/1257), [#1234](https://github.com/LykosAI/StabilityMatrix/issues/1234) - "no such file or directory" errors when updating certain packages after folder migration
28+
- Fixed [#1274](https://github.com/LykosAI/StabilityMatrix/issues/1274), [#1276](https://github.com/LykosAI/StabilityMatrix/issues/1276) - incorrect torch installed when updating to InvokeAI v5.12+
2829
- Fixed missing shared folder links for SwarmUI's diffusion_models and clip folders
2930
### Supporters
3031
#### 🌟 Visionaries

StabilityMatrix.Core/Models/Packages/InvokeAI.cs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -145,13 +145,13 @@ public override async Task InstallPackage(
145145
progress?.Report(new ProgressReport(-1f, "Installing Package", isIndeterminate: true));
146146

147147
var torchVersion = options.PythonOptions.TorchIndex ?? GetRecommendedTorchVersion();
148-
var isBlackwell =
149-
SettingsManager.Settings.PreferredGpu?.IsBlackwellGpu() ?? HardwareHelper.HasBlackwellGpu();
148+
var isLegacyNvidiaGpu =
149+
SettingsManager.Settings.PreferredGpu?.IsLegacyNvidiaGpu() ?? HardwareHelper.HasLegacyNvidiaGpu();
150150
var index = torchVersion switch
151151
{
152152
TorchIndex.Cpu => "https://download.pytorch.org/whl/cpu",
153-
TorchIndex.Cuda when isBlackwell => "https://download.pytorch.org/whl/cu128",
154-
TorchIndex.Cuda => "https://download.pytorch.org/whl/cu126",
153+
TorchIndex.Cuda when isLegacyNvidiaGpu => "https://download.pytorch.org/whl/cu126",
154+
TorchIndex.Cuda => "https://download.pytorch.org/whl/cu128",
155155
TorchIndex.Rocm => "https://download.pytorch.org/whl/rocm6.2.4",
156156
TorchIndex.Mps => "https://download.pytorch.org/whl/cpu",
157157
_ => string.Empty,

0 commit comments

Comments
 (0)