Ionite
1 year ago
committed by
GitHub
12 changed files with 261 additions and 70 deletions
@ -0,0 +1,122 @@ |
|||||||
|
using System.Diagnostics; |
||||||
|
using System.Text.RegularExpressions; |
||||||
|
using StabilityMatrix.Core.Helper; |
||||||
|
using StabilityMatrix.Core.Helper.Cache; |
||||||
|
using StabilityMatrix.Core.Models.Progress; |
||||||
|
using StabilityMatrix.Core.Processes; |
||||||
|
using StabilityMatrix.Core.Services; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Core.Models.Packages; |
||||||
|
|
||||||
|
public class Fooocus : BaseGitPackage |
||||||
|
{ |
||||||
|
public Fooocus(IGithubApiCache githubApi, ISettingsManager settingsManager, |
||||||
|
IDownloadService downloadService, IPrerequisiteHelper prerequisiteHelper) : base(githubApi, |
||||||
|
settingsManager, downloadService, prerequisiteHelper) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
public override string Name => "Fooocus"; |
||||||
|
public override string DisplayName { get; set; } = "Fooocus"; |
||||||
|
public override string Author => "lllyasviel"; |
||||||
|
|
||||||
|
public override string Blurb => |
||||||
|
"Fooocus is a rethinking of Stable Diffusion and Midjourney’s designs"; |
||||||
|
|
||||||
|
public override string LicenseType => "GPL-3.0"; |
||||||
|
public override string LicenseUrl => "https://github.com/lllyasviel/Fooocus/blob/main/LICENSE"; |
||||||
|
public override string LaunchCommand => "launch.py"; |
||||||
|
|
||||||
|
public override Uri PreviewImageUri => |
||||||
|
new("https://user-images.githubusercontent.com/19834515/261830306-f79c5981-cf80-4ee3-b06b-3fef3f8bfbc7.png"); |
||||||
|
|
||||||
|
public override List<LaunchOptionDefinition> LaunchOptions => new() |
||||||
|
{ |
||||||
|
LaunchOptionDefinition.Extras |
||||||
|
}; |
||||||
|
|
||||||
|
public override Dictionary<SharedFolderType, IReadOnlyList<string>> SharedFolders => new() |
||||||
|
{ |
||||||
|
[SharedFolderType.StableDiffusion] = new[] {"models/checkpoints"}, |
||||||
|
[SharedFolderType.Diffusers] = new[] {"models/diffusers"}, |
||||||
|
[SharedFolderType.Lora] = new[] {"models/loras"}, |
||||||
|
[SharedFolderType.CLIP] = new[] {"models/clip"}, |
||||||
|
[SharedFolderType.TextualInversion] = new[] {"models/embeddings"}, |
||||||
|
[SharedFolderType.VAE] = new[] {"models/vae"}, |
||||||
|
[SharedFolderType.ApproxVAE] = new[] {"models/vae_approx"}, |
||||||
|
[SharedFolderType.ControlNet] = new[] {"models/controlnet"}, |
||||||
|
[SharedFolderType.GLIGEN] = new[] {"models/gligen"}, |
||||||
|
[SharedFolderType.ESRGAN] = new[] {"models/upscale_models"}, |
||||||
|
[SharedFolderType.Hypernetwork] = new[] {"models/hypernetworks"} |
||||||
|
}; |
||||||
|
|
||||||
|
public override async Task<string> GetLatestVersion() |
||||||
|
{ |
||||||
|
var release = await GetLatestRelease().ConfigureAwait(false); |
||||||
|
return release.TagName!; |
||||||
|
} |
||||||
|
|
||||||
|
public override async Task InstallPackage(IProgress<ProgressReport>? progress = null) |
||||||
|
{ |
||||||
|
await base.InstallPackage(progress).ConfigureAwait(false); |
||||||
|
var venvRunner = await SetupVenv(InstallLocation).ConfigureAwait(false); |
||||||
|
|
||||||
|
progress?.Report(new ProgressReport(-1f, "Installing torch...", isIndeterminate: true)); |
||||||
|
|
||||||
|
var torchVersion = "cpu"; |
||||||
|
var gpus = HardwareHelper.IterGpuInfo().ToList(); |
||||||
|
|
||||||
|
if (gpus.Any(g => g.IsNvidia)) |
||||||
|
{ |
||||||
|
torchVersion = "cu118"; |
||||||
|
} |
||||||
|
else if (HardwareHelper.PreferRocm()) |
||||||
|
{ |
||||||
|
torchVersion = "rocm5.4.2"; |
||||||
|
} |
||||||
|
|
||||||
|
await venvRunner |
||||||
|
.PipInstall( |
||||||
|
$"torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/{torchVersion}", |
||||||
|
OnConsoleOutput).ConfigureAwait(false); |
||||||
|
|
||||||
|
progress?.Report(new ProgressReport(-1f, "Installing requirements...", |
||||||
|
isIndeterminate: true)); |
||||||
|
await venvRunner.PipInstall("-r requirements_versions.txt", OnConsoleOutput) |
||||||
|
.ConfigureAwait(false); |
||||||
|
} |
||||||
|
|
||||||
|
public override async Task RunPackage(string installedPackagePath, string command, string arguments) |
||||||
|
{ |
||||||
|
await SetupVenv(installedPackagePath).ConfigureAwait(false); |
||||||
|
|
||||||
|
void HandleConsoleOutput(ProcessOutput s) |
||||||
|
{ |
||||||
|
OnConsoleOutput(s); |
||||||
|
|
||||||
|
if (s.Text.Contains("To see the GUI go to", StringComparison.OrdinalIgnoreCase)) |
||||||
|
{ |
||||||
|
var regex = new Regex(@"(https?:\/\/)([^:\s]+):(\d+)"); |
||||||
|
var match = regex.Match(s.Text); |
||||||
|
if (match.Success) |
||||||
|
{ |
||||||
|
WebUrl = match.Value; |
||||||
|
} |
||||||
|
OnStartupComplete(WebUrl); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void HandleExit(int i) |
||||||
|
{ |
||||||
|
Debug.WriteLine($"Venv process exited with code {i}"); |
||||||
|
OnExit(i); |
||||||
|
} |
||||||
|
|
||||||
|
var args = $"\"{Path.Combine(installedPackagePath, command)}\" {arguments}"; |
||||||
|
|
||||||
|
VenvRunner?.RunDetached( |
||||||
|
args.TrimEnd(), |
||||||
|
HandleConsoleOutput, |
||||||
|
HandleExit); |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue