From c839ba59dde8a9bfae70eec4c7d77cf132ff5698 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Mon, 29 Jan 2024 21:55:02 -0800 Subject: [PATCH] Update legacy_preprocessors.py --- .../scripts/legacy_preprocessors.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py b/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py index 6f4d9fa0..4c512fe5 100644 --- a/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py +++ b/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py @@ -91,11 +91,11 @@ class LegacyPreprocessor(Preprocessor): del slider_3 if self.unload_function is not None or self.managed_model is not None: - context = automatic_memory_management() + context = automatic_memory_management else: - context = contextlib.nullcontext() + context = contextlib.nullcontext - with context: + with context(): result, is_image = self.call_function(img=input_image, res=resolution, thr_a=slider_1, thr_b=slider_2, **kwargs) if is_image: