suppress A matching Triton is not available message

you can all now stop worrying about it

1 files changed, 3 insertions, 1 deletions

diff --git a/webui.py b/webui.py
index bc2baeab..e1565a8d 100644
--- a/webui.py
+++ b/webui.py
@@ -1,6 +1,5 @@
 import os

 import sys

-import threading

 import time

 import importlib

 import signal

@@ -10,6 +9,9 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.gzip import GZipMiddleware

 from packaging import version

 

+import logging

+logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())

+

 from modules import import_hook, errors, extra_networks

 from modules import extra_networks_hypernet, ui_extra_networks_hypernets, ui_extra_networks_textual_inversion

 from modules.call_queue import wrap_queued_call, queue_lock, wrap_gradio_gpu_call