From 4d85f61650ed9b128024859d10709d9f5b17eaf6 Mon Sep 17 00:00:00 2001 From: leejet Date: Sun, 14 Dec 2025 00:57:20 +0800 Subject: [PATCH 1/2] fix: add lora info to image metadata --- examples/cli/main.cpp | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/examples/cli/main.cpp b/examples/cli/main.cpp index 417d211aa..6010789ed 100644 --- a/examples/cli/main.cpp +++ b/examples/cli/main.cpp @@ -232,7 +232,14 @@ static std::string sd_basename(const std::string& path) { } std::string get_image_params(const SDCliParams& cli_params, const SDContextParams& ctx_params, const SDGenerationParams& gen_params, int64_t seed) { - std::string parameter_string = gen_params.prompt + "\n"; + std::string parameter_string = gen_params.prompt; + for (const auto& item : gen_params.lora_map) { + parameter_string += ""; + } + for (const auto& item : gen_params.high_noise_lora_map) { + parameter_string += ""; + } + parameter_string += "\n"; if (gen_params.negative_prompt.size() != 0) { parameter_string += "Negative prompt: " + gen_params.negative_prompt + "\n"; } From 292b0bd6e8d4258a7bd04e7b40e96ee20a45a237 Mon Sep 17 00:00:00 2001 From: leejet Date: Sun, 14 Dec 2025 01:08:25 +0800 Subject: [PATCH 2/2] fix --- examples/cli/main.cpp | 9 +-------- examples/common/common.hpp | 2 ++ 2 files changed, 3 insertions(+), 8 deletions(-) diff --git a/examples/cli/main.cpp b/examples/cli/main.cpp index 6010789ed..22480d7eb 100644 --- a/examples/cli/main.cpp +++ b/examples/cli/main.cpp @@ -232,14 +232,7 @@ static std::string sd_basename(const std::string& path) { } std::string get_image_params(const SDCliParams& cli_params, const SDContextParams& ctx_params, const SDGenerationParams& gen_params, int64_t seed) { - std::string parameter_string = gen_params.prompt; - for (const auto& item : gen_params.lora_map) { - parameter_string += ""; - } - for (const auto& item : gen_params.high_noise_lora_map) { - parameter_string += ""; - } - parameter_string += "\n"; + std::string parameter_string = gen_params.prompt_with_lora + "\n"; if (gen_params.negative_prompt.size() != 0) { parameter_string += "Negative prompt: " + gen_params.negative_prompt + "\n"; } diff --git a/examples/common/common.hpp b/examples/common/common.hpp index ccd01cec4..bf38379d2 100644 --- a/examples/common/common.hpp +++ b/examples/common/common.hpp @@ -863,6 +863,7 @@ static bool is_absolute_path(const std::string& p) { struct SDGenerationParams { std::string prompt; + std::string prompt_with_lora; // for metadata record only std::string negative_prompt; int clip_skip = -1; // <= 0 represents unspecified int width = 512; @@ -1476,6 +1477,7 @@ struct SDGenerationParams { } bool process_and_check(SDMode mode, const std::string& lora_model_dir) { + prompt_with_lora = prompt; if (width <= 0) { fprintf(stderr, "error: the width must be greater than 0\n"); return false;