fix pix2pix producing bad results

5e80d9ee · AUTOMATIC1111 · 47bccbeb · 5e80d9ee
Commit 5e80d9ee authored Nov 07, 2023 by AUTOMATIC1111
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/processing.py modules/processing.py +1 -1

No files found.
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -296,7 +296,7 @@ class StableDiffusionProcessing:
        return conditioning

    def edit_image_conditioning(self, source_image):
-        conditioning_image = images_tensor_to_samples(source_image*0.5+0.5, approximation_indexes.get(opts.sd_vae_encode_method))
+        conditioning_image = shared.sd_model.encode_first_stage(source_image).mode()

        return conditioning_image