diff --git a/main.py b/main.py index 9b2a33011fda..b23d508168bd 100644 --- a/main.py +++ b/main.py @@ -112,6 +112,7 @@ def execute_script(script_path): if os.name == "nt": + os.environ['MIMALLOC_PURGE_DELAY'] = '0' logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage()) if __name__ == "__main__":