From b750c0d7c3ca027299e3cd7a3ea19377638a9691 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jens=20S=C3=BClwald?= Date: Mon, 12 Dec 2022 16:24:03 +0100 Subject: [PATCH] The exception should also mention dpm2 --- ui/sd_internal/runtime.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py index d3fd6e86..2bf53d0c 100644 --- a/ui/sd_internal/runtime.py +++ b/ui/sd_internal/runtime.py @@ -822,7 +822,7 @@ def _txt2img(opt_W, opt_H, opt_n_samples, opt_ddim_steps, opt_scale, start_code, move_to_cpu(thread_data.modelCS) if thread_data.test_sd2 and sampler_name not in ('plms', 'ddim', 'dpm2'): - raise Exception('Only plms and ddim samplers are supported right now, in SD 2.0') + raise Exception('Only plms, ddim and dpm2 samplers are supported right now, in SD 2.0') # samples, _ = sampler.sample(S=opt.steps,