diff --git a/invokeai/backend/stable_diffusion/extensions/base.py b/invokeai/backend/stable_diffusion/extensions/base.py index 802af86e6d..835fe0aaf9 100644 --- a/invokeai/backend/stable_diffusion/extensions/base.py +++ b/invokeai/backend/stable_diffusion/extensions/base.py @@ -52,7 +52,7 @@ class ExtensionBase: return self._callbacks @contextmanager - def patch_extension(self, context: DenoiseContext): + def patch_extension(self, ctx: DenoiseContext): yield None @contextmanager diff --git a/invokeai/backend/stable_diffusion/extensions_manager.py b/invokeai/backend/stable_diffusion/extensions_manager.py index 1cae2e4219..a9e554ad98 100644 --- a/invokeai/backend/stable_diffusion/extensions_manager.py +++ b/invokeai/backend/stable_diffusion/extensions_manager.py @@ -52,13 +52,13 @@ class ExtensionsManager: cb.function(ctx) @contextmanager - def patch_extensions(self, context: DenoiseContext): + def patch_extensions(self, ctx: DenoiseContext): if self._is_canceled and self._is_canceled(): raise CanceledException with ExitStack() as exit_stack: for ext in self._extensions: - exit_stack.enter_context(ext.patch_extension(context)) + exit_stack.enter_context(ext.patch_extension(ctx)) yield None