Hide Triton warning on Windows

Triton doesnt work on Windows so hide the error
2026-01-11 14:50:49 +08:00 · 2023-02-16 18:19:26 +00:00 · 2023-02-16 18:19:26 +00:00 · d11f81521c
commit d11f81521c
parent 2b8a5e8a42
1 changed files with 4 additions and 0 deletions
--- 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