瀏覽代碼

fix pix2pix producing bad results

AUTOMATIC1111 1 年之前
父節點
當前提交
5e80d9ee99
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      modules/processing.py

+ 1 - 1
modules/processing.py

@@ -296,7 +296,7 @@ class StableDiffusionProcessing:
         return conditioning
         return conditioning
 
 
     def edit_image_conditioning(self, source_image):
     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
         return conditioning_image