Add check for WSL_DISTRO_NAME for is_wsl()

WSL supports custom kernels. Add check for WSL_DISTRO_NAME to check custom kernels.
2026-03-01 13:56:19 +08:00 · 2026-03-01 13:56:19 +08:00 · 6ff2489510
parent 850e8b42ff
commit 6ff2489510
1 changed files with 2 additions and 0 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -186,6 +186,8 @@ def is_wsl():
        return True
    elif version.endswith("microsoft-standard-WSL2"):
        return True
+    if os.environ.get("WSL_DISTRO_NAME"):
+        return True
    return False

 def get_torch_device():