feat: add flag to disable the embedding of generation metadata

Author: wbruna

examples/cli/README.md

@@ -123,6 +123,7 @@ Generation Options:
   --vace-strength <float>                  wan vace strength
   --increase-ref-index                     automatically increase the indices of references images based on the order they are listed (starting with 1).
   --disable-auto-resize-ref-image          disable auto resize of ref images
+  --disable-image-metadata                 do not embed generation metadata on image files
   -s, --seed                               RNG seed (default: 42, use random seed for < 0)
   --sampling-method                        sampling method, one of [euler, euler_a, heun, dpm2, dpm++2s_a, dpm++2m, dpm++2mv2, ipndm, ipndm_v, lcm, ddim_trailing, tcd,
                                            res_multistep, res_2s] (default: euler for Flux/SD3/Wan, euler_a otherwise)

examples/cli/main.cpp

@@ -357,12 +357,14 @@ bool save_results(const SDCliParams& cli_params,
         if (!img.data)
             return false;
 
-        std::string params = get_image_params(ctx_params, gen_params, gen_params.seed + idx);
+        std::string params = gen_params.embed_image_metadata
+            ? get_image_params(ctx_params, gen_params, gen_params.seed + idx)
+            : "";
         int ok             = 0;
         if (is_jpg) {
-            ok = stbi_write_jpg(path.string().c_str(), img.width, img.height, img.channel, img.data, 90, params.c_str());
+            ok = stbi_write_jpg(path.string().c_str(), img.width, img.height, img.channel, img.data, 90, params.size() > 0 ? params.c_str() : nullptr);
         } else {
-            ok = stbi_write_png(path.string().c_str(), img.width, img.height, img.channel, img.data, 0, params.c_str());
+            ok = stbi_write_png(path.string().c_str(), img.width, img.height, img.channel, img.data, 0, params.size() > 0 ? params.c_str() : nullptr);
         }
         LOG_INFO("save result image %d to '%s' (%s)", idx, path.string().c_str(), ok ? "success" : "failure");
         return ok != 0;

examples/common/common.hpp

@@ -1042,6 +1042,7 @@ struct SDGenerationParams {
     std::string control_video_path;
     bool auto_resize_ref_image = true;
     bool increase_ref_index    = false;
+    bool embed_image_metadata  = true;
 
     std::vector<int> skip_layers = {7, 8, 9};
     sd_sample_params_t sample_params;
@@ -1267,6 +1268,11 @@ struct SDGenerationParams {
              "disable auto resize of ref images",
              false,
              &auto_resize_ref_image},
+            {"",
+             "--disable-image-metadata",
+             "do not embed generation metadata on image files",
+             false,
+             &embed_image_metadata},
         };
 
         auto on_seed_arg = [&](int argc, const char** argv, int index) {
@@ -1597,6 +1603,7 @@ struct SDGenerationParams {
 
         load_if_exists("auto_resize_ref_image", auto_resize_ref_image);
         load_if_exists("increase_ref_index", increase_ref_index);
+        load_if_exists("embed_image_metadata", embed_image_metadata);
 
         load_if_exists("skip_layers", skip_layers);
         load_if_exists("high_noise_skip_layers", high_noise_skip_layers);

examples/server/README.md

@@ -115,6 +115,7 @@ Default Generation Options:
   --vace-strength <float>                  wan vace strength
   --increase-ref-index                     automatically increase the indices of references images based on the order they are listed (starting with 1).
   --disable-auto-resize-ref-image          disable auto resize of ref images
+  --disable-image-metadata                 do not embed generation metadata on image files
   -s, --seed                               RNG seed (default: 42, use random seed for < 0)
   --sampling-method                        sampling method, one of [euler, euler_a, heun, dpm2, dpm++2s_a, dpm++2m, dpm++2mv2, ipndm, ipndm_v, lcm, ddim_trailing, tcd,
                                            res_multistep, res_2s] (default: euler for Flux/SD3/Wan, euler_a otherwise)

examples/server/main.cpp

@@ -256,7 +256,7 @@ std::vector<uint8_t> write_image_to_vector(
             result = stbi_write_jpg_to_func(c_func, &ctx, width, height, channels, image, quality);
             break;
         case ImageFormat::PNG:
-            result = stbi_ext_write_png_to_func(c_func, &ctx, width, height, channels, image, width * channels, params.c_str());
+            result = stbi_ext_write_png_to_func(c_func, &ctx, width, height, channels, image, width * channels, params.size() > 0 ? params.c_str() : nullptr);
             break;
         default:
             throw std::runtime_error("invalid image format");
@@ -528,7 +528,9 @@ int main(int argc, const char** argv) {
                 if (results[i].data == nullptr) {
                     continue;
                 }
-                std::string params = get_image_params(ctx_params, gen_params, gen_params.seed + i);
+                std::string params = gen_params.embed_image_metadata
+                    ? get_image_params(ctx_params, gen_params, gen_params.seed + i)
+                    : "";
                 auto image_bytes = write_image_to_vector(output_format == "jpeg" ? ImageFormat::JPEG : ImageFormat::PNG,
                                                          results[i].data,
                                                          results[i].width,
@@ -775,7 +777,9 @@ int main(int argc, const char** argv) {
             for (int i = 0; i < num_results; i++) {
                 if (results[i].data == nullptr)
                     continue;
-                std::string params = get_image_params(ctx_params, gen_params, gen_params.seed + i);
+                std::string params = gen_params.embed_image_metadata
+                    ? get_image_params(ctx_params, gen_params, gen_params.seed + i)
+                    : "";
                 auto image_bytes = write_image_to_vector(output_format == "jpeg" ? ImageFormat::JPEG : ImageFormat::PNG,
                                                          results[i].data,
                                                          results[i].width,
@@ -1093,7 +1097,9 @@ int main(int argc, const char** argv) {
                     continue;
                 }
 
-                std::string params = get_image_params(ctx_params, gen_params, gen_params.seed + i);
+                std::string params = gen_params.embed_image_metadata
+                    ? get_image_params(ctx_params, gen_params, gen_params.seed + i)
+                    : "";
                 auto image_bytes = write_image_to_vector(ImageFormat::PNG,
                                                          results[i].data,
                                                          results[i].width,