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

Minor fixes

Browse Source
This commit is contained in:
Sergey Borisov 2024-07-17 03:48:37 +03:00
parent 137202b77c
commit 79e35bd0d3
3 changed files with 75 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/invocations/denoise_latents.py
View File

@ -723,8 +723,6 @@ class DenoiseLatentsInvocation(BaseInvocation):
@torch.no_grad() @torch.no_grad()
@SilenceWarnings() # This quenches the NSFW nag from diffusers. @SilenceWarnings() # This quenches the NSFW nag from diffusers.
def _new_invoke(self, context: InvocationContext) -> LatentsOutput: def _new_invoke(self, context: InvocationContext) -> LatentsOutput:
# TODO: remove supression when extensions which use models added
with ExitStack() as exit_stack: # noqa: F841
ext_manager = ExtensionsManager() ext_manager = ExtensionsManager()
device = TorchDevice.choose_torch_device() device = TorchDevice.choose_torch_device()
@ -782,15 +780,15 @@ class DenoiseLatentsInvocation(BaseInvocation):
scheduler=scheduler, scheduler=scheduler,
) )
# get the unet's config so that we can pass the base to sd_step_callback()
unet_config = context.models.get_config(self.unet.unet.key)
### preview ### preview
def step_callback(state: PipelineIntermediateState) -> None: def step_callback(state: PipelineIntermediateState) -> None:
context.util.sd_step_callback(state, unet_config.base) context.util.sd_step_callback(state, unet_config.base)
ext_manager.add_extension(PreviewExt(step_callback)) ext_manager.add_extension(PreviewExt(step_callback))
# get the unet's config so that we can pass the base to sd_step_callback()
unet_config = context.models.get_config(self.unet.unet.key)
# ext: t2i/ip adapter # ext: t2i/ip adapter
ext_manager.callbacks.setup(denoise_ctx, ext_manager) ext_manager.callbacks.setup(denoise_ctx, ext_manager)

8
invokeai/backend/model_patcher.py
View File

@ -43,11 +43,11 @@ class ModelPatcher:
processor (Type[Any]): Class which will be initialized for each key and passed to set_attn_processor(...). processor (Type[Any]): Class which will be initialized for each key and passed to set_attn_processor(...).
""" """
unet_orig_processors = unet.attn_processors unet_orig_processors = unet.attn_processors
try:
# create separate instance for each attention, to be able modify each attention separately
new_attn_processors = {key: processor_cls() for key in unet_orig_processors.keys()}
unet.set_attn_processor(new_attn_processors)
# create separate instance for each attention, to be able modify each attention separately
unet_new_processors = {key: processor_cls() for key in unet_orig_processors.keys()}
try:
unet.set_attn_processor(unet_new_processors)
yield None yield None
finally: finally:

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

@ -8,8 +8,6 @@ from typing import TYPE_CHECKING, Callable, Dict
import torch import torch
from diffusers import UNet2DConditionModel from diffusers import UNet2DConditionModel
from invokeai.backend.util.devices import TorchDevice
if TYPE_CHECKING: if TYPE_CHECKING:
from invokeai.backend.stable_diffusion.denoise_context import DenoiseContext from invokeai.backend.stable_diffusion.denoise_context import DenoiseContext
from invokeai.backend.stable_diffusion.extensions import ExtensionBase from invokeai.backend.stable_diffusion.extensions import ExtensionBase