change StableDiffusionProcessing to internally use sampler name instead of sampler index

1 files changed, 2 insertions, 2 deletions

diff --git a/modules/img2img.py b/modules/img2img.py
index be9f3653..9fc5b693 100644
--- a/modules/img2img.py
+++ b/modules/img2img.py
@@ -6,7 +6,7 @@ import traceback
 import numpy as np

 from PIL import Image, ImageOps, ImageChops

 

-from modules import devices

+from modules import devices, sd_samplers

 from modules.processing import Processed, StableDiffusionProcessingImg2Img, process_images

 from modules.shared import opts, state

 import modules.shared as shared

@@ -99,7 +99,7 @@ def img2img(mode: int, prompt: str, negative_prompt: str, prompt_style: str, pro
         seed_resize_from_h=seed_resize_from_h,

         seed_resize_from_w=seed_resize_from_w,

         seed_enable_extras=seed_enable_extras,

-        sampler_index=sampler_index,

+        sampler_index=sd_samplers.samplers_for_img2img[sampler_index].name,

         batch_size=batch_size,

         n_iter=n_iter,

         steps=steps,