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

Fixed problem with inpainting after controlnet support added to main.

Browse Source 
Problem was that controlnet support involved adding **kwargs to method calls down in denoising loop, and AddsMaskLatents didn't accept **kwarg arg. So just changed to accept and pass on **kwargs.
This commit is contained in:
user1 2023-05-30 00:26:07 -07:00 committed by Kent Keirsey
parent 070218aba7
commit b1b94a3d56
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -75,10 +75,10 @@ class AddsMaskLatents:
initial_image_latents: torch.Tensor initial_image_latents: torch.Tensor
def __call__( def __call__(
self, latents: torch.Tensor, t: torch.Tensor, text_embeddings: torch.Tensor self, latents: torch.Tensor, t: torch.Tensor, text_embeddings: torch.Tensor, **kwargs,
) -> torch.Tensor: ) -> torch.Tensor:
model_input = self.add_mask_channels(latents) model_input = self.add_mask_channels(latents)
return self.forward(model_input, t, text_embeddings) return self.forward(model_input, t, text_embeddings, **kwargs)
def add_mask_channels(self, latents): def add_mask_channels(self, latents):
batch_size = latents.size(0) batch_size = latents.size(0)