From 78e917a6fb245678d970db8151cb10844db5d686 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Fri, 3 Feb 2023 21:40:03 +0530
Subject: [PATCH] Fix the broken 'Make Similar Images' button

---
 ui/easydiffusion/renderer.py | 2 ++
 ui/media/js/engine.js        | 4 ++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/ui/easydiffusion/renderer.py b/ui/easydiffusion/renderer.py
index 836ac934..672614ba 100644
--- a/ui/easydiffusion/renderer.py
+++ b/ui/easydiffusion/renderer.py
@@ -65,6 +65,8 @@ def generate_images_internal(req: GenerateImageRequest, task_data: TaskData, dat
     callback = make_step_callback(req, task_data, data_queue, task_temp_images, step_callback, stream_image_progress)
 
     try:
+        if req.init_image is not None: req.sampler_name = 'ddim'
+
         images = generate_images(context, callback=callback, **req.dict())
         user_stopped = False
     except UserInitiatedStop:
diff --git a/ui/media/js/engine.js b/ui/media/js/engine.js
index cec2eb36..0d51fdf4 100644
--- a/ui/media/js/engine.js
+++ b/ui/media/js/engine.js
@@ -718,7 +718,7 @@
         "height": 'number',
         "seed": 'number',
 
-        "sampler": 'string',
+        "sampler_name": 'string',
         "use_stable_diffusion_model": 'string',
         "num_inference_steps": 'number',
         "guidance_scale": 'number',
@@ -731,7 +731,7 @@
         "output_quality": 'number',
     }
     const TASK_DEFAULTS = {
-        "sampler": "plms",
+        "sampler_name": "plms",
         "use_stable_diffusion_model": "sd-v1-4",
         "num_inference_steps": 50,
         "guidance_scale": 7.5,