Skip to content

Commit 6056f72

Browse files
committed
Add Win-ROCm launch notice injected into the console output on package launch
1 parent 533fbe8 commit 6056f72

1 file changed

Lines changed: 11 additions & 0 deletions

File tree

StabilityMatrix.Core/Models/Packages/OneTrainer.cs

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -180,13 +180,24 @@ await SetupVenv(installLocation, pythonVersion: PyVersion.Parse(installedPackage
180180
);
181181
}
182182

183+
foreach (var line in GetLaunchNoticeLines(installedPackage))
184+
{
185+
onConsoleOutput?.Invoke(ProcessOutput.FromStdOutLine($"{line}{Environment.NewLine}"));
186+
}
187+
183188
VenvRunner.RunDetached(
184189
[Path.Combine(installLocation, options.Command ?? LaunchCommand), .. options.Arguments],
185190
onConsoleOutput,
186191
OnExit
187192
);
188193
}
189194

195+
private IReadOnlyList<string> GetLaunchNoticeLines(InstalledPackage installedPackage)
196+
{
197+
var selectedTorchIndex = installedPackage.PreferredTorchIndex ?? GetRecommendedTorchVersion();
198+
return rocmPackageHelper.GetWindowsLaunchNoticeLines(selectedTorchIndex);
199+
}
200+
190201
public override List<LaunchOptionDefinition> LaunchOptions => [LaunchOptionDefinition.Extras];
191202
public override Dictionary<SharedFolderType, IReadOnlyList<string>>? SharedFolders { get; }
192203
public override Dictionary<SharedOutputType, IReadOnlyList<string>>? SharedOutputFolders { get; }

0 commit comments

Comments
 (0)