aboutsummaryrefslogtreecommitdiff
path: root/modules/processing.py
diff options
context:
space:
mode:
authorezxzeng <zhixuan.zeng@gmail.com>2023-11-07 15:25:27 -0500
committerGitHub <noreply@github.com>2023-11-07 15:25:27 -0500
commitf9c14a8c8c8b682c2a7f7a11306b26e2a28ebeec (patch)
tree1c7f6abc11bd2aacd2a0d16a0b7db1963c9b26cd /modules/processing.py
parent759515316e8ec536f34fad616e8c6a33674a164b (diff)
parent5e80d9ee99c5899e5e2b130408ffb65a0585a62a (diff)
Merge branch 'dev' into ui_mobile_optimizations
Diffstat (limited to 'modules/processing.py')
-rw-r--r--modules/processing.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/modules/processing.py b/modules/processing.py
index 40598f5c..b0e240a4 100644
--- 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
@@ -886,6 +886,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
devices.torch_gc()
+ state.nextjob()
+
if p.scripts is not None:
p.scripts.postprocess_batch(p, x_samples_ddim, batch_number=n)
@@ -958,8 +960,6 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
devices.torch_gc()
- state.nextjob()
-
if not infotexts:
infotexts.append(Processed(p, []).infotext(p, 0))