From 67e59b9ceb45c88c6ec6d6fdac445c0620fb1488 Mon Sep 17 00:00:00 2001 From: receyuki Date: Fri, 29 Sep 2023 02:36:03 +0800 Subject: [PATCH] Fix scheduler_str --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index f893cac..5140d83 100644 --- a/nodes.py +++ b/nodes.py @@ -278,7 +278,7 @@ class SDPromptSaver: model_name_real = model_name_str if model_name_str else model_name sampler_name_real = sampler_name_str if sampler_name_str else sampler_name - scheduler_real = scheduler if scheduler_str else scheduler + scheduler_real = scheduler_str if scheduler_str else scheduler for image in images: i = 255.0 * image.cpu().numpy()