|
|
|
|
@ -14,7 +14,7 @@ using StabilityMatrix.Core.Services;
|
|
|
|
|
namespace StabilityMatrix.Core.Models.Packages; |
|
|
|
|
|
|
|
|
|
[Singleton(typeof(BasePackage))] |
|
|
|
|
public class StableDiffusionDirectMl : BaseGitPackage |
|
|
|
|
public class StableDiffusionDirectMl : A3WebUI |
|
|
|
|
{ |
|
|
|
|
private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); |
|
|
|
|
|
|
|
|
|
@ -44,145 +44,15 @@ public class StableDiffusionDirectMl : BaseGitPackage
|
|
|
|
|
) |
|
|
|
|
: base(githubApi, settingsManager, downloadService, prerequisiteHelper) { } |
|
|
|
|
|
|
|
|
|
public override Dictionary<SharedFolderType, IReadOnlyList<string>> SharedFolders => |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
[SharedFolderType.StableDiffusion] = new[] { "models/Stable-diffusion" }, |
|
|
|
|
[SharedFolderType.ESRGAN] = new[] { "models/ESRGAN" }, |
|
|
|
|
[SharedFolderType.RealESRGAN] = new[] { "models/RealESRGAN" }, |
|
|
|
|
[SharedFolderType.SwinIR] = new[] { "models/SwinIR" }, |
|
|
|
|
[SharedFolderType.Lora] = new[] { "models/Lora" }, |
|
|
|
|
[SharedFolderType.LyCORIS] = new[] { "models/LyCORIS" }, |
|
|
|
|
[SharedFolderType.ApproxVAE] = new[] { "models/VAE-approx" }, |
|
|
|
|
[SharedFolderType.VAE] = new[] { "models/VAE" }, |
|
|
|
|
[SharedFolderType.DeepDanbooru] = new[] { "models/deepbooru" }, |
|
|
|
|
[SharedFolderType.Karlo] = new[] { "models/karlo" }, |
|
|
|
|
[SharedFolderType.TextualInversion] = new[] { "embeddings" }, |
|
|
|
|
[SharedFolderType.Hypernetwork] = new[] { "models/hypernetworks" }, |
|
|
|
|
[SharedFolderType.ControlNet] = new[] { "models/ControlNet" }, |
|
|
|
|
[SharedFolderType.Codeformer] = new[] { "models/Codeformer" }, |
|
|
|
|
[SharedFolderType.LDSR] = new[] { "models/LDSR" }, |
|
|
|
|
[SharedFolderType.AfterDetailer] = new[] { "models/adetailer" } |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
public override Dictionary<SharedOutputType, IReadOnlyList<string>>? SharedOutputFolders => |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
[SharedOutputType.Extras] = new[] { "outputs/extras-images" }, |
|
|
|
|
[SharedOutputType.Saved] = new[] { "log/images" }, |
|
|
|
|
[SharedOutputType.Img2Img] = new[] { "outputs/img2img-images" }, |
|
|
|
|
[SharedOutputType.Text2Img] = new[] { "outputs/txt2img-images" }, |
|
|
|
|
[SharedOutputType.Img2ImgGrids] = new[] { "outputs/img2img-grids" }, |
|
|
|
|
[SharedOutputType.Text2ImgGrids] = new[] { "outputs/txt2img-grids" } |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
[SuppressMessage("ReSharper", "ArrangeObjectCreationWhenTypeNotEvident")] |
|
|
|
|
public override List<LaunchOptionDefinition> LaunchOptions => |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "Host", |
|
|
|
|
Type = LaunchOptionType.String, |
|
|
|
|
DefaultValue = "localhost", |
|
|
|
|
Options = new() { "--server-name" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "Port", |
|
|
|
|
Type = LaunchOptionType.String, |
|
|
|
|
DefaultValue = "7860", |
|
|
|
|
Options = new() { "--port" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "VRAM", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
InitialValue = HardwareHelper |
|
|
|
|
.IterGpuInfo() |
|
|
|
|
.Select(gpu => gpu.MemoryLevel) |
|
|
|
|
.Max() switch |
|
|
|
|
{ |
|
|
|
|
Level.Low => "--lowvram", |
|
|
|
|
Level.Medium => "--medvram", |
|
|
|
|
_ => null |
|
|
|
|
}, |
|
|
|
|
Options = new() { "--lowvram", "--medvram", "--medvram-sdxl" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "Xformers", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
InitialValue = HardwareHelper.HasNvidiaGpu(), |
|
|
|
|
Options = new() { "--xformers" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "API", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
InitialValue = true, |
|
|
|
|
Options = new() { "--api" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "Auto Launch Web UI", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
InitialValue = false, |
|
|
|
|
Options = new() { "--autolaunch" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "Skip Torch CUDA Check", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
InitialValue = !HardwareHelper.HasNvidiaGpu(), |
|
|
|
|
Options = new() { "--skip-torch-cuda-test" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "Skip Python Version Check", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
InitialValue = true, |
|
|
|
|
Options = new() { "--skip-python-version-check" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "No Half", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
Description = "Do not switch the model to 16-bit floats", |
|
|
|
|
InitialValue = HardwareHelper.HasAmdGpu(), |
|
|
|
|
Options = new() { "--no-half" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "Skip SD Model Download", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
InitialValue = false, |
|
|
|
|
Options = new() { "--no-download-sd-model" } |
|
|
|
|
}, |
|
|
|
|
new() |
|
|
|
|
{ |
|
|
|
|
Name = "Skip Install", |
|
|
|
|
Type = LaunchOptionType.Bool, |
|
|
|
|
Options = new() { "--skip-install" } |
|
|
|
|
}, |
|
|
|
|
LaunchOptionDefinition.Extras |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
public override IEnumerable<SharedFolderMethod> AvailableSharedFolderMethods => |
|
|
|
|
new[] { SharedFolderMethod.Symlink, SharedFolderMethod.None }; |
|
|
|
|
|
|
|
|
|
public override IEnumerable<TorchVersion> AvailableTorchVersions => |
|
|
|
|
new[] { TorchVersion.Cpu, TorchVersion.DirectMl }; |
|
|
|
|
|
|
|
|
|
public override Task<string> GetLatestVersion() => Task.FromResult("master"); |
|
|
|
|
|
|
|
|
|
public override bool ShouldIgnoreReleases => true; |
|
|
|
|
|
|
|
|
|
public override string OutputFolderName => "outputs"; |
|
|
|
|
|
|
|
|
|
public override async Task InstallPackage( |
|
|
|
|
string installLocation, |
|
|
|
|
TorchVersion torchVersion, |
|
|
|
|
SharedFolderMethod selectedSharedFolderMethod, |
|
|
|
|
DownloadPackageVersionOptions versionOptions, |
|
|
|
|
IProgress<ProgressReport>? progress = null, |
|
|
|
|
Action<ProcessOutput>? onConsoleOutput = null |
|
|
|
|
@ -205,6 +75,8 @@ public class StableDiffusionDirectMl : BaseGitPackage
|
|
|
|
|
break; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
await venvRunner.PipInstall("httpx==0.24.1").ConfigureAwait(false); |
|
|
|
|
|
|
|
|
|
// Install requirements file |
|
|
|
|
progress?.Report( |
|
|
|
|
new ProgressReport(-1f, "Installing Package Requirements", isIndeterminate: true) |
|
|
|
|
@ -218,34 +90,4 @@ public class StableDiffusionDirectMl : BaseGitPackage
|
|
|
|
|
|
|
|
|
|
progress?.Report(new ProgressReport(1f, "Install complete", isIndeterminate: false)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public override async Task RunPackage( |
|
|
|
|
string installedPackagePath, |
|
|
|
|
string command, |
|
|
|
|
string arguments, |
|
|
|
|
Action<ProcessOutput>? onConsoleOutput |
|
|
|
|
) |
|
|
|
|
{ |
|
|
|
|
await SetupVenv(installedPackagePath).ConfigureAwait(false); |
|
|
|
|
|
|
|
|
|
void HandleConsoleOutput(ProcessOutput s) |
|
|
|
|
{ |
|
|
|
|
onConsoleOutput?.Invoke(s); |
|
|
|
|
|
|
|
|
|
if (!s.Text.Contains("Running on", StringComparison.OrdinalIgnoreCase)) |
|
|
|
|
return; |
|
|
|
|
|
|
|
|
|
var regex = new Regex(@"(https?:\/\/)([^:\s]+):(\d+)"); |
|
|
|
|
var match = regex.Match(s.Text); |
|
|
|
|
if (!match.Success) |
|
|
|
|
return; |
|
|
|
|
|
|
|
|
|
WebUrl = match.Value; |
|
|
|
|
OnStartupComplete(WebUrl); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
var args = $"\"{Path.Combine(installedPackagePath, command)}\" {arguments}"; |
|
|
|
|
|
|
|
|
|
VenvRunner.RunDetached(args.TrimEnd(), HandleConsoleOutput, OnExit); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
JT
1 year ago
committed by
GitHub