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.Cpu => "cpu",
TorchVersion.Cuda => "cu121", TorchVersion.Cuda => "cu121",
TorchVersion.Rocm => "rocm5.6", TorchVersion.Rocm => "rocm5.6",
TorchVersion.Mps => "nightly/cpu", TorchVersion.Mps => "cpu",
_ => throw new ArgumentOutOfRangeException(nameof(torchVersion), torchVersion, null) _ => throw new ArgumentOutOfRangeException(nameof(torchVersion), torchVersion, null)
} }
) )

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

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

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

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

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

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

Loading…
Cancel
Save