Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use consistent param names in patch_extension(...) functions: context -> ctx.

Browse Source
This commit is contained in:
Ryan Dick 2024-07-23 09:18:04 -04:00
parent 4e8dcb7a1a
commit 39e804d0f8
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/stable_diffusion/extensions/base.py
View File

@ -52,7 +52,7 @@ class ExtensionBase:
return self._callbacks return self._callbacks
@contextmanager @contextmanager
def patch_extension(self, context: DenoiseContext): def patch_extension(self, ctx: DenoiseContext):
yield None yield None
@contextmanager @contextmanager

4
invokeai/backend/stable_diffusion/extensions_manager.py
View File

@ -52,13 +52,13 @@ class ExtensionsManager:
cb.function(ctx) cb.function(ctx)
@contextmanager @contextmanager
def patch_extensions(self, context: DenoiseContext): def patch_extensions(self, ctx: DenoiseContext):
if self._is_canceled and self._is_canceled(): if self._is_canceled and self._is_canceled():
raise CanceledException raise CanceledException
with ExitStack() as exit_stack: with ExitStack() as exit_stack:
for ext in self._extensions: for ext in self._extensions:
exit_stack.enter_context(ext.patch_extension(context)) exit_stack.enter_context(ext.patch_extension(ctx))
yield None yield None