From 768cfe3aaba096dd618e783fc5d5ec35bd26b200 Mon Sep 17 00:00:00 2001 From: user1 Date: Fri, 26 May 2023 13:02:45 -0700 Subject: [PATCH] Core implementation of ControlNet and MultiControlNet. --- invokeai/backend/stable_diffusion/diffusers_pipeline.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/invokeai/backend/stable_diffusion/diffusers_pipeline.py b/invokeai/backend/stable_diffusion/diffusers_pipeline.py index b5e686ec64..41146c8004 100644 --- a/invokeai/backend/stable_diffusion/diffusers_pipeline.py +++ b/invokeai/backend/stable_diffusion/diffusers_pipeline.py @@ -803,7 +803,7 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline): noise: torch.Tensor, run_id=None, callback=None, - ) -> InvokeAIStableDiffusionPipelineOutput: + ) -> InvokeAIStableDiffusionPipelineOutput: timesteps, _ = self.get_img2img_timesteps(num_inference_steps, strength) result_latents, result_attention_maps = self.latents_from_embeddings( latents=initial_latents if strength < 1.0 else torch.zeros_like( @@ -1031,7 +1031,6 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline): dtype=torch.float16, do_classifier_free_guidance=True, ): - if not isinstance(image, torch.Tensor): if isinstance(image, PIL.Image.Image): image = [image]