Add extra_info to the Prompt Saver

This commit is contained in:
Rhys Yang
2023-10-17 23:26:02 +08:00
parent f67ae16743
commit baceed2a01
+5
View File
@@ -248,6 +248,7 @@ class SDPromptSaver:
"STRING",
{"default": "%H%M%S", "multiline": False},
),
"extra_info": ("STRING", {"default": "", "multiline": True}),
},
"hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
}
@@ -283,6 +284,7 @@ class SDPromptSaver:
jpg_webp_quality: int = 100,
date_format: str = "%Y-%m-%d",
time_format: str = "%H%M%S",
extra_info: str = "",
prompt=None,
extra_pnginfo=None,
):
@@ -302,6 +304,8 @@ class SDPromptSaver:
sampler_name_real = sampler_name_str if sampler_name_str else sampler_name
scheduler_real = scheduler_str if scheduler_str else scheduler
extra_info_real = f", Extra info: {extra_info}" if extra_info else ""
variable_map = {
"%date": self.get_time(date_format),
"%time": self.get_time(time_format),
@@ -336,6 +340,7 @@ class SDPromptSaver:
f"{model_hash}"
f"Model: {Path(model_name_real).stem}, "
f"Version: ComfyUI"
f"{extra_info_real}"
)
subfolder = self.get_path(path, variable_map)
output_folder = Path(full_output_folder) / subfolder