diff --git a/StabilityMatrix.Core/Models/Packages/ComfyZluda.cs b/StabilityMatrix.Core/Models/Packages/ComfyZluda.cs index 72911e1d..cb45b7a6 100644 --- a/StabilityMatrix.Core/Models/Packages/ComfyZluda.cs +++ b/StabilityMatrix.Core/Models/Packages/ComfyZluda.cs @@ -27,6 +27,7 @@ IPyInstallationManager pyInstallationManager "https://github.com/lshqqytiger/ZLUDA/releases/download/rel.5e717459179dc272b7d7d23391f0fad66c7459cf/ZLUDA-nightly-windows-rocm6-amd64.zip"; private const string HipSdkExtensionDownloadUrl = "https://cdn.lykos.ai/HIP-SDK-extension.7z"; + private const string VenvDirectoryName = "venv"; private Process? zludaProcess; @@ -202,6 +203,11 @@ private Dictionary GetEnvVars(bool isInstall) ["GIT"] = portableGitBin.JoinFile("git.exe"), }; + if (isInstall) + { + envVars["VIRTUAL_ENV"] = VenvDirectoryName; + } + if (envVars.TryGetValue("PATH", out var pathValue)) { envVars["PATH"] = Compat.GetEnvPathWithExtensions(hipBinPath, portableGitBin, pathValue);