Processing only, no CFGDenoiser change

Allows instruct-pix2pix
3 years ago · 3b2ad20ac1
parent cf0cfefe91
commit 3b2ad20ac1
1 changed files with 1 additions and 1 deletions
--- a/modules/processing.py
+++ b/modules/processing.py
@ -186,7 +186,7 @@ class StableDiffusionProcessing:
        return conditioning

    def edit_image_conditioning(self, source_image):
-        conditioning_image = self.sd_model.get_first_stage_encoding(self.sd_model.encode_first_stage(source_image))
+        conditioning_image = self.sd_model.encode_first_stage(source_image).mode()

        return conditioning_image