fix wrong torch being installed

mohnjiles · mohnjiles · commit f5d45995e0d1 · 2025-08-24T13:36:07.000-07:00
diff --git a/StabilityMatrix.Core/Models/Packages/AiToolkit.cs b/StabilityMatrix.Core/Models/Packages/AiToolkit.cs
@@ -98,9 +98,10 @@ public override async Task InstallPackage(
         pipArgs = new PipInstallArgs("--upgrade")
             .WithParsedFromRequirementsTxt(
                 await requirements.ReadAllTextAsync(cancellationToken).ConfigureAwait(false),
-                excludePattern: "torch$|numpy"
+                excludePattern: "torch"
             )
-            .AddArg(Compat.IsWindows ? "triton-windows" : "triton");
+            .AddArg(Compat.IsWindows ? "triton-windows" : "triton")
+            .WithTorchExtraIndex(isBlackwell ? "cu128" : "cu126");
 
         if (installedPackage.PipOverrides != null)
         {

Original file line number	Diff line number	Diff line change
`@@ -98,9 +98,10 @@ public override async Task InstallPackage(`
`98`	`98`	`pipArgs = new PipInstallArgs("--upgrade")`
`99`	`99`	`.WithParsedFromRequirementsTxt(`
`100`	`100`	`await requirements.ReadAllTextAsync(cancellationToken).ConfigureAwait(false),`
`101`		`- excludePattern: "torch$\|numpy"`
	`101`	`+ excludePattern: "torch"`
`102`	`102`	`)`
`103`		`- .AddArg(Compat.IsWindows ? "triton-windows" : "triton");`
	`103`	`+ .AddArg(Compat.IsWindows ? "triton-windows" : "triton")`
	`104`	`+ .WithTorchExtraIndex(isBlackwell ? "cu128" : "cu126");`
`104`	`105`
`105`	`106`	`if (installedPackage.PipOverrides != null)`
`106`	`107`	`{`