Fix FluxGym, kohya, framepack installs, and update invoke difficulty

Author: mohnjiles

StabilityMatrix.Core/Models/Packages/FluxGym.cs

@@ -113,10 +113,12 @@ await PrerequisiteHelper
         var config = new PipInstallConfig
         {
             RequirementsFilePaths = ["sd-scripts/requirements.txt", "requirements.txt"],
-            RequirementsExcludePattern = "(torch|torchvision|torchaudio|xformers|bitsandbytes)",
+            RequirementsExcludePattern =
+                "(diffusers\\[torch\\]==0.32.1|torch|torchvision|torchaudio|xformers|bitsandbytes|-e\\s\\.)",
             TorchaudioVersion = " ",
             CudaIndex = isLegacyNvidiaGpu ? "cu126" : "cu128",
             ExtraPipArgs = ["bitsandbytes>=0.46.0"],
+            PostInstallPipArgs = ["diffusers[torch]==0.32.1"],
         };
 
         await StandardPipInstallProcessAsync(
@@ -129,6 +131,8 @@ await StandardPipInstallProcessAsync(
                 cancellationToken
             )
             .ConfigureAwait(false);
+
+        await venvRunner.PipInstall(["-e", "./sd-scripts"], onConsoleOutput).ConfigureAwait(false);
     }
 
     public override async Task RunPackage(

StabilityMatrix.Core/Models/Packages/FramePack.cs

@@ -119,6 +119,7 @@ public override async Task InstallPackage(
             CudaIndex = isLegacyNvidia ? "cu126" : "cu128",
             UpgradePackages = true,
             ExtraPipArgs = extraArgs,
+            PostInstallPipArgs = ["numpy==1.26.4"],
         };
 
         await StandardPipInstallProcessAsync(

StabilityMatrix.Core/Models/Packages/InvokeAI.cs

@@ -40,7 +40,7 @@ IPyInstallationManager pyInstallationManager
 
     public override string Blurb => "Professional Creative Tools for Stable Diffusion";
     public override string LaunchCommand => "invokeai-web";
-    public override PackageDifficulty InstallerSortOrder => PackageDifficulty.Nightmare;
+    public override PackageDifficulty InstallerSortOrder => PackageDifficulty.Advanced;
 
     public override Uri PreviewImageUri =>
         new("https://raw.githubusercontent.com/invoke-ai/InvokeAI/main/docs/assets/canvas_preview.png");

StabilityMatrix.Core/Models/Packages/KohyaSs.cs

@@ -170,7 +170,8 @@ await venvRunner
                 XformersVersion = " ",
                 CudaIndex = isLegacyNvidia ? "cu126" : "cu128",
                 // Add back the generic bitsandbytes and the specific numpy version
-                ExtraPipArgs = ["bitsandbytes", "numpy==1.26.4"],
+                ExtraPipArgs = ["bitsandbytes"],
+                PostInstallPipArgs = ["numpy==1.26.4"],
             };
 
             await StandardPipInstallProcessAsync(