You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.7 KiB
73 lines
2.7 KiB
12 months ago
|
using System.Diagnostics;
|
||
|
|
using System.Text.RegularExpressions;
|
||
|
|
using StabilityMatrix.Core.Attributes;
|
||
|
|
using StabilityMatrix.Core.Helper;
|
||
|
|
using StabilityMatrix.Core.Helper.Cache;
|
||
|
|
using StabilityMatrix.Core.Models.FileInterfaces;
|
||
|
|
using StabilityMatrix.Core.Models.Progress;
|
||
|
|
using StabilityMatrix.Core.Processes;
|
||
|
|
using StabilityMatrix.Core.Python;
|
||
|
|
using StabilityMatrix.Core.Services;
|
||
|
|
|
||
|
|
namespace StabilityMatrix.Core.Models.Packages;
|
||
|
|
|
||
|
|
[Singleton(typeof(BasePackage))]
|
||
|
|
public class RuinedFooocus : Fooocus
|
||
|
|
{
|
||
|
|
public RuinedFooocus(
|
||
|
|
IGithubApiCache githubApi,
|
||
|
|
ISettingsManager settingsManager,
|
||
|
|
IDownloadService downloadService,
|
||
|
|
IPrerequisiteHelper prerequisiteHelper
|
||
|
|
)
|
||
|
|
: base(githubApi, settingsManager, downloadService, prerequisiteHelper) { }
|
||
|
|
|
||
|
|
public override string Name => "RuinedFooocus";
|
||
|
|
public override string DisplayName { get; set; } = "RuinedFooocus";
|
||
|
|
public override string Author => "runew0lf";
|
||
|
|
public override string Blurb =>
|
||
|
|
"RuinedFooocus combines the best aspects of Stable Diffusion and Midjourney into one seamless, cutting-edge experience";
|
||
|
|
public override string LicenseUrl =>
|
||
|
|
"https://github.com/runew0lf/RuinedFooocus/blob/main/LICENSE";
|
||
|
|
public override Uri PreviewImageUri =>
|
||
|
|
new("https://raw.githubusercontent.com/runew0lf/pmmconfigs/main/RuinedFooocus_ss.png");
|
||
|
|
public override PackageDifficulty InstallerSortOrder => PackageDifficulty.Expert;
|
||
|
|
|
||
|
|
public override async Task InstallPackage(
|
||
|
|
string installLocation,
|
||
|
|
TorchVersion torchVersion,
|
||
|
|
SharedFolderMethod selectedSharedFolderMethod,
|
||
|
|
DownloadPackageVersionOptions versionOptions,
|
||
|
|
IProgress<ProgressReport>? progress = null,
|
||
|
|
Action<ProcessOutput>? onConsoleOutput = null
|
||
|
|
)
|
||
|
|
{
|
||
|
|
if (torchVersion == TorchVersion.Cuda)
|
||
|
|
{
|
||
|
|
var venvRunner = await SetupVenv(installLocation, forceRecreate: true)
|
||
|
|
.ConfigureAwait(false);
|
||
|
|
|
||
|
|
progress?.Report(new ProgressReport(-1f, "Installing torch...", isIndeterminate: true));
|
||
|
|
|
||
|
|
await InstallCudaTorch(venvRunner, progress, onConsoleOutput).ConfigureAwait(false);
|
||
|
|
|
||
|
|
var requirements = new FilePath(installLocation, "requirements_versions.txt");
|
||
|
|
await venvRunner
|
||
|
|
.PipInstallFromRequirements(requirements, onConsoleOutput, excludes: "torch")
|
||
|
|
.ConfigureAwait(false);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
await base.InstallPackage(
|
||
|
|
installLocation,
|
||
|
|
torchVersion,
|
||
|
|
selectedSharedFolderMethod,
|
||
|
|
versionOptions,
|
||
|
|
progress,
|
||
|
|
onConsoleOutput
|
||
|
|
)
|
||
|
|
.ConfigureAwait(false);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|