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

Renames

Browse Source
This commit is contained in:
Sergey Borisov
2024-07-12 22:44:00 +03:00
parent bd8ae5d896
commit 3a9dda9177
4 changed files with 19 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -16,6 +16,10 @@ if TYPE_CHECKING:
class ExtCallbacksApi(ABC):
@abstractmethod
def setup(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
pass
@abstractmethod
def pre_denoise_loop(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
pass
@ -33,15 +37,15 @@ class ExtCallbacksApi(ABC):
pass
@abstractmethod
def modify_noise_prediction(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
def pre_unet(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
pass
@abstractmethod
def pre_unet_forward(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
def post_unet(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
pass
@abstractmethod
def pre_unet_load(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
def post_apply_cfg(self, ctx: DenoiseContext, ext_manager: ExtensionsManager):
pass