Hide Triton warning on Windows

Triton doesnt work on Windows so hide the error

Hide Triton warning on Windows
Triton doesnt work on Windows so hide the error
07872996 · pythongosssss · GitHub · 56498d50 · 07872996
Unverified Commit 07872996 authored Feb 16, 2023 by pythongosssss Committed by GitHub Feb 16, 2023
Show 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