Skip to content

Commit 4516c34

Browse files
authored
Merge pull request LykosAI#1524 from NeuralFault/ComfyUI-WinROCm-update
ComfyUI: Updated gfx110X Windows ROCm nightly index. Added Win-ROCm EnVars
2 parents 895b247 + 965ad7a commit 4516c34

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

StabilityMatrix.Core/Models/Packages/ComfyUI.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,7 @@ await StandardPipInstallProcessAsync(
382382
var indexUrl = gfxArch switch
383383
{
384384
"gfx1151" => "https://rocm.nightlies.amd.com/v2/gfx1151",
385-
_ when gfxArch.StartsWith("gfx110") => "https://rocm.nightlies.amd.com/v2/gfx110X-dgpu",
385+
_ when gfxArch.StartsWith("gfx110") => "https://rocm.nightlies.amd.com/v2/gfx110X-all",
386386
_ when gfxArch.StartsWith("gfx120") => "https://rocm.nightlies.amd.com/v2/gfx120X-all",
387387
_ => throw new ArgumentOutOfRangeException(
388388
nameof(gfxArch),
@@ -870,6 +870,8 @@ private ImmutableDictionary<string, string> GetEnvVars(ImmutableDictionary<strin
870870
// set some experimental speed improving env vars for Windows ROCm
871871
return env.SetItem("PYTORCH_TUNABLEOP_ENABLED", "1")
872872
.SetItem("MIOPEN_FIND_MODE", "2")
873-
.SetItem("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", "1");
873+
.SetItem("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", "1")
874+
.SetItem("PYTORCH_ALLOC_CONF", "max_split_size_mb:6144,garbage_collection_threshold:0.8") // greatly helps prevent GPU OOM and instability/driver timeouts/OS hard locks and decreases dependency on Tiled VAE at standard res's
875+
.SetItem("COMFYUI_USE_MIOPEN", "1"); // re-enables "cudnn" in ComfyUI as it's needed for MiOpen to function properly
874876
}
875877
}

0 commit comments

Comments
 (0)