diff --git a/ldm/invoke/__init__.py b/ldm/invoke/__init__.py index 33c6d891dd..b643e80240 100644 --- a/ldm/invoke/__init__.py +++ b/ldm/invoke/__init__.py @@ -2,3 +2,12 @@ from ._version import __version__ __app_id__= 'invoke-ai/InvokeAI' __app_name__= 'InvokeAI' + + +def _ignore_xformers_triton_message_on_windows(): + import logging + logging.getLogger("xformers").addFilter( + lambda record: 'A matching Triton is not available' not in record.getMessage()) + +# In order to be effective, this needs to happen before anything could possibly import xformers. +_ignore_xformers_triton_message_on_windows()