From bab9dba0d9deee2980689f5076f6ff2359f71c1c Mon Sep 17 00:00:00 2001
From: Rhys Yang <receyuki@gmail.com>
Date: Thu, 19 Oct 2023 05:23:36 +0800
Subject: [PATCH] Remove model_name_str, sampler_name_str and scheduler_str

---
 nodes.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/nodes.py b/nodes.py
index 48066eb..17c2687 100644
--- a/nodes.py
+++ b/nodes.py
@@ -201,7 +201,7 @@ class SDPromptSaver:
                 ),
                 "path": ("STRING", {"default": "%date/", "multiline": False}),
                 "model_name": (folder_paths.get_filename_list("checkpoints"),),
-                "model_name_str": ("STRING", {"default": ""}),
+                # "model_name_str": ("STRING", {"default": ""}),
                 "seed": (
                     "INT",
                     {
@@ -225,9 +225,9 @@ class SDPromptSaver:
                     },
                 ),
                 "sampler_name": (comfy.samplers.KSampler.SAMPLERS,),
-                "sampler_name_str": ("STRING", {"default": ""}),
+                # "sampler_name_str": ("STRING", {"default": ""}),
                 "scheduler": (comfy.samplers.KSampler.SCHEDULERS,),
-                "scheduler_str": ("STRING", {"default": ""}),
+                # "scheduler_str": ("STRING", {"default": ""}),
                 "width": (
                     "INT",
                     {"default": 1, "min": 1, "max": MAX_RESOLUTION, "step": 8},
@@ -302,6 +302,8 @@ class SDPromptSaver:
 
         results = list()
 
+        # model_name_str, sampler_name_str, scheduler_str = None, None, None
+
         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_str if scheduler_str else scheduler