From f92f9586827b9fc73b69b9890b1b88d057470fca Mon Sep 17 00:00:00 2001
From: Will Miao <willmiao.dev@gmail.com>
Date: Tue, 28 Jul 2026 21:39:09 +0800
Subject: [PATCH] fix(SaveImageLM): correct scheduler mapping and deduplicate
 sampler map

- Fix incorrect mapping: "normal" -> "Normal" (was "Simple")
- Replace inline sampler_mapping with CIVITAI_SAMPLER_MAP reference
  to eliminate duplicate definition
---
 py/nodes/save_image.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/py/nodes/save_image.py b/py/nodes/save_image.py
index 6d3ff2aa..8b0532a4 100644
--- a/py/nodes/save_image.py
+++ b/py/nodes/save_image.py
@@ -446,7 +446,16 @@ class SaveImageLM:
                 lora_resource["versionName"] = lora_civitai["name"]
             civitai_resources.append(lora_resource)
 
-        sampler_display = self._get_civitai_sampler_name(sampler, scheduler)
+        sampler_name = CIVITAI_SAMPLER_MAP.get(sampler, sampler) if sampler else None
+
+        scheduler_mapping = {
+            "normal": "Normal",
+            "karras": "Karras",
+            "exponential": "Exponential",
+            "sgm_uniform": "SGM Uniform",
+            "sgm_quadratic": "SGM Quadratic",
+        }
+        scheduler_name = scheduler_mapping.get(scheduler, scheduler) if scheduler else None
 
         # Build output lines
         lines = [prompt] if prompt else [""]
@@ -456,8 +465,11 @@ class SaveImageLM:
         params: list[str] = []
         if steps is not None:
             params.append(f"Steps: {steps}")
-        if sampler_display:
-            params.append(f"Sampler: {sampler_display}")
+        if sampler_name:
+            if scheduler_name:
+                params.append(f"Sampler: {sampler_name} {scheduler_name}")
+            else:
+                params.append(f"Sampler: {sampler_name}")
         if cfg is not None:
             params.append(f"CFG scale: {cfg}")
         if seed is not None: