Hide Triton warning on Windows

Triton doesnt work on Windows so hide the error
2023-02-16 18:19:26 +00:00 · 2023-02-16 18:19:26 +00:00 · 0787299610
parent 56498d505a
commit 0787299610
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