From 3e7f14af2c65eaf2d3a589008906c3e7138b832b Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Wed, 1 Feb 2023 11:50:27 +0530 Subject: [PATCH] Don't use Rich Tracebacks, can cause a memory leak. It keeps a reference to the Exception object (which in turn keeps references to any torch Tensors in the stack, preventing their garbage-collection) --- ui/easydiffusion/app.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ui/easydiffusion/app.py b/ui/easydiffusion/app.py index 2a8d0804..63505ed3 100644 --- a/ui/easydiffusion/app.py +++ b/ui/easydiffusion/app.py @@ -20,7 +20,7 @@ logging.basicConfig( level=logging.INFO, format=LOG_FORMAT, datefmt="%X", - handlers=[RichHandler(markup=True, rich_tracebacks=True, show_time=False, show_level=False)] + handlers=[RichHandler(markup=True, rich_tracebacks=False, show_time=False, show_level=False)], ) SD_DIR = os.getcwd()