Browse Source

Remove preview torch from mps install

Stable supports MPS now, pre flag getting dev builds instead

(cherry picked from commit 6e079ae84e0523c360115fba8e9efaa993981544)
pull/629/head
Ionite 6 months ago
parent
commit
93a16cbd3e
No known key found for this signature in database
  1. 2
      StabilityMatrix.Core/Models/Packages/A3WebUI.cs
  2. 3
      StabilityMatrix.Core/Models/Packages/ComfyUI.cs
  3. 2
      StabilityMatrix.Core/Models/Packages/SDWebForge.cs
  4. 2
      StabilityMatrix.Core/Models/Packages/StableDiffusionUx.cs

2
StabilityMatrix.Core/Models/Packages/A3WebUI.cs

@ -223,7 +223,7 @@ public class A3WebUI(
TorchVersion.Cpu => "cpu",
TorchVersion.Cuda => "cu121",
TorchVersion.Rocm => "rocm5.6",
TorchVersion.Mps => "nightly/cpu",
TorchVersion.Mps => "cpu",
_ => throw new ArgumentOutOfRangeException(nameof(torchVersion), torchVersion, null)
}
)

3
StabilityMatrix.Core/Models/Packages/ComfyUI.cs

@ -209,8 +209,7 @@ public class ComfyUI(
pipArgs = torchVersion switch
{
TorchVersion.DirectMl => pipArgs.WithTorchDirectML(),
TorchVersion.Mps
=> pipArgs.AddArg("--pre").WithTorch().WithTorchVision().WithTorchExtraIndex("nightly/cpu"),
TorchVersion.Mps => pipArgs.WithTorch().WithTorchVision().WithTorchExtraIndex("cpu"),
_
=> pipArgs
.AddArg("--upgrade")

2
StabilityMatrix.Core/Models/Packages/SDWebForge.cs

@ -162,7 +162,7 @@ public class SDWebForge(
TorchVersion.Cpu => "cpu",
TorchVersion.Cuda => "cu121",
TorchVersion.Rocm => "rocm5.6",
TorchVersion.Mps => "nightly/cpu",
TorchVersion.Mps => "cpu",
_ => throw new ArgumentOutOfRangeException(nameof(torchVersion), torchVersion, null)
}
);

2
StabilityMatrix.Core/Models/Packages/StableDiffusionUx.cs

@ -211,7 +211,7 @@ public class StableDiffusionUx(
new PipInstallArgs()
.WithTorch("==2.1.2")
.WithTorchVision()
.WithTorchExtraIndex("nightly/cpu"),
.WithTorchExtraIndex("cpu"),
onConsoleOutput
)
.ConfigureAwait(false);

Loading…
Cancel
Save