Merge branch 'hide-triton-error-windows' of https://github.com/pythongosssss/ComfyUI

1cf5b612 · comfyanonymous · 638f9245 · 07872996 · 1cf5b612
Commit 1cf5b612 authored Feb 17, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

main.py main.py +4 -0

No files found.
--- a/main.py
+++ b/main.py
@@ -7,6 +7,10 @@ import heapq
 import traceback
 import asyncio

+if os.name == "nt":
+    import logging
+    logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
+
 try:
    import aiohttp
    from aiohttp import web