diff --git a/scripts/webui.py b/scripts/webui.py index 35548b1..91d1b54 100644 --- a/scripts/webui.py +++ b/scripts/webui.py @@ -1779,7 +1779,7 @@ def imgproc(image,image_batch,imgproc_prompt,imgproc_toggles, imgproc_upscale_to elif sampler_name == 'k_dpm_2_a': sampler = KDiffusionSampler(model,'dpm_2_ancestral') elif sampler_name == 'k_dpm_2': - sampler_name = KDiffusionSampler(model,'dpm_2') + sampler = KDiffusionSampler(model,'dpm_2') elif sampler_name == 'k_euler_a': sampler = KDiffusionSampler(model,'euler_ancestral') elif sampler_name == 'k_euler':