Warn user if imatrix will get overwritten

Author: stduhpf

examples/cli/main.cpp

@@ -376,6 +376,16 @@ bool parse_options(int argc, const char** argv, ArgOptions& options) {
     return true;
 }
 
+int file_exists(const char *path) {
+    FILE *file;
+    if ((file = fopen(path, "r"))) {
+        fclose(file);
+        return 1;
+    }
+    return 0;
+}
+
+
 void parse_args(int argc, const char** argv, SDParams& params) {
     ArgOptions options;
     options.string_options = {
@@ -663,6 +673,14 @@ void parse_args(int argc, const char** argv, SDParams& params) {
     if (!isfinite(params.img_cfg_scale)) {
         params.img_cfg_scale = params.cfg_scale;
     }
+
+    if (params.imatrix_out.size() > 0 && file_exists(params.imatrix_out.c_str())) {
+        // imatrix file already exists
+        if (std::find(params.imatrix_in.begin(), params.imatrix_in.end(), params.imatrix_out) == params.imatrix_in.end()) {
+            printf("\n IMPORTANT: imatrix file %s already exists, but wasn't found in the imatrix inputs.\n", params.imatrix_out.c_str());
+            printf("%s will get overwritten!\n", params.imatrix_out.c_str());
+        }
+    }
 }
 
 static std::string sd_basename(const std::string& path) {

stable-diffusion.h

@@ -243,6 +243,7 @@ SD_API void free_upscaler_ctx(upscaler_ctx_t* upscaler_ctx);
 
 SD_API sd_image_t upscale(upscaler_ctx_t* upscaler_ctx, sd_image_t input_image, uint32_t upscale_factor);
 
+SD_API void setConvertImatrixCollector(void * collector);
 SD_API bool convert(const char* model_path, const char* clip_l_path, const char* clip_g_path, const char* t5xxl_path, const char* diffusion_model_path,
                     const char* vae_path,
                     const char* output_path,