Skip to content

Commit 0046e9d

Browse files
authored
fix merge conflict
1 parent 6049cb3 commit 0046e9d

File tree

1 file changed

+22
-49
lines changed

1 file changed

+22
-49
lines changed

StabilityMatrix.Core/Models/Packages/KohyaSs.cs

Lines changed: 22 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -125,17 +125,14 @@ await PrerequisiteHelper
125125
await using var venvRunner = await SetupVenvPure(installLocation).ConfigureAwait(false);
126126

127127
// Extra dep needed before running setup since v23.0.x
128-
var pipArgs = new PipInstallArgs("rich", "packaging");
128+
var pipArgs = new PipInstallArgs("rich", "packaging", "setuptools", "uv");
129129
if (installedPackage.PipOverrides != null)
130130
{
131131
pipArgs = pipArgs.WithUserOverrides(installedPackage.PipOverrides);
132132
}
133133

134134
await venvRunner.PipInstall(pipArgs).ConfigureAwait(false);
135135

136-
<<<<<<< HEAD
137-
if (Compat.IsWindows)
138-
=======
139136
var isLegacyNvidia =
140137
SettingsManager.Settings.PreferredGpu?.IsLegacyNvidiaGpu() ?? HardwareHelper.HasLegacyNvidiaGpu();
141138
var torchExtraIndex = isLegacyNvidia ? "cu126" : "cu128";
@@ -150,68 +147,44 @@ await PrerequisiteHelper
150147
.AddArg("--force-reinstall");
151148

152149
if (installedPackage.PipOverrides != null)
153-
>>>>>>> c422920b (Merge pull request #1076 from ionite34/fix-legacy-gpu)
154150
{
155-
await venvRunner
156-
.CustomInstall(["setup/setup_windows.py", "--headless"], onConsoleOutput)
157-
.ConfigureAwait(false);
151+
pipArgs = pipArgs.WithUserOverrides(installedPackage.PipOverrides);
158152
}
159-
else if (Compat.IsLinux)
153+
154+
await venvRunner.PipInstall(pipArgs, onConsoleOutput).ConfigureAwait(false);
155+
156+
if (Compat.IsLinux)
160157
{
161158
await venvRunner
162159
.CustomInstall(
163160
[
164161
"setup/setup_linux.py",
165162
"--platform-requirements-file=requirements_linux.txt",
166-
"--no_run_accelerate"
163+
"--no_run_accelerate",
167164
],
168165
onConsoleOutput
169166
)
170167
.ConfigureAwait(false);
168+
pipArgs = new PipInstallArgs();
171169
}
172-
173-
var isBlackwell =
174-
SettingsManager.Settings.PreferredGpu?.IsBlackwellGpu() ?? HardwareHelper.HasBlackwellGpu();
175-
176-
if (isBlackwell)
170+
else if (Compat.IsWindows)
177171
{
172+
var requirements = new FilePath(installLocation, "requirements_windows.txt");
178173
pipArgs = new PipInstallArgs()
179-
.WithTorch()
180-
.WithTorchVision()
181-
.WithTorchAudio()
182-
.WithTorchExtraIndex("cu128")
183-
.AddArg("--force-reinstall");
184-
185-
if (installedPackage.PipOverrides != null)
186-
{
187-
pipArgs = pipArgs.WithUserOverrides(installedPackage.PipOverrides);
188-
}
189-
190-
await venvRunner.PipInstall(pipArgs, onConsoleOutput).ConfigureAwait(false);
191-
192-
pipArgs = new PipInstallArgs()
193-
.AddArg("--pre")
194-
.AddArg("-U")
195-
.AddArg("--no-deps")
196-
.AddArg("xformers");
197-
198-
if (installedPackage.PipOverrides != null)
199-
{
200-
pipArgs = pipArgs.WithUserOverrides(installedPackage.PipOverrides);
201-
}
202-
203-
await venvRunner.PipInstall(pipArgs, onConsoleOutput).ConfigureAwait(false);
204-
205-
pipArgs = new PipInstallArgs().AddArg("-U").AddArg("bitsandbytes");
206-
207-
if (installedPackage.PipOverrides != null)
208-
{
209-
pipArgs = pipArgs.WithUserOverrides(installedPackage.PipOverrides);
210-
}
174+
.WithParsedFromRequirementsTxt(
175+
await requirements.ReadAllTextAsync(cancellationToken).ConfigureAwait(false),
176+
"bitsandbytes==0\\.44\\.0"
177+
)
178+
.AddArg("bitsandbytes");
179+
}
211180

212-
await venvRunner.PipInstall(pipArgs, onConsoleOutput).ConfigureAwait(false);
213-
await venvRunner.PipInstall("numpy==1.26.4", onConsoleOutput).ConfigureAwait(false);
181+
if (installedPackage.PipOverrides != null)
182+
{
183+
pipArgs = pipArgs.WithUserOverrides(installedPackage.PipOverrides);
214184
}
185+
186+
await venvRunner.PipInstall(pipArgs, onConsoleOutput).ConfigureAwait(false);
187+
await venvRunner.PipInstall("numpy==1.26.4", onConsoleOutput).ConfigureAwait(false);
215188
}
216189

217190
public override async Task RunPackage(

0 commit comments

Comments
 (0)