From 1fb6b228507fc146a40695cc1ebc0f97a6bee6de Mon Sep 17 00:00:00 2001
From: Cyberhan123 <38837039+Cyberhan123@users.noreply.github.com>
Date: Sat, 13 Jun 2026 13:08:14 +0800
Subject: [PATCH] feat: add free_sd_images function to manage memory for C API
 (#1633)

---
 include/stable-diffusion.h |  4 ++++
 src/stable-diffusion.cpp   | 15 +++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/include/stable-diffusion.h b/include/stable-diffusion.h
index 17596f84..2175f895 100644
--- a/include/stable-diffusion.h
+++ b/include/stable-diffusion.h
@@ -491,6 +491,10 @@ SD_API bool preprocess_canny(sd_image_t image,
 SD_API const char* sd_commit(void);
 SD_API const char* sd_version(void);
 
+// for C API, caller needs to call free_sd_images to free the memory after use
+// This helps avoid CRT problems on Windows when memory is allocated in the library but freed in the caller, which may use a different CRT.
+SD_API void free_sd_images(sd_image_t* result_images, int num_images);
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/src/stable-diffusion.cpp b/src/stable-diffusion.cpp
index e98a0327..916a93f2 100644
--- a/src/stable-diffusion.cpp
+++ b/src/stable-diffusion.cpp
@@ -5561,3 +5561,18 @@ SD_API bool generate_video(sd_ctx_t* sd_ctx,
     }
     return true;
 }
+
+SD_API void free_sd_images(sd_image_t* result_images, int num_images) {
+    if (result_images == nullptr) {
+        return;
+    }
+
+    for (int i = 0; i < num_images; ++i) {
+        if (result_images[i].data != nullptr) {
+            free(result_images[i].data);
+            result_images[i].data = nullptr;
+        }
+    }
+
+    free(result_images);
+}
\ No newline at end of file