refactor(tx): add version util

thxCode · thxCode · commit 33518e210e10 · 2024-12-29T17:01:43.000+08:00
Signed-off-by: thxCode &lt;thxcode0824@gmail.com&gt;
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -2208,3 +2208,10 @@ SD_API sd_image_t* img2vid(sd_ctx_t* sd_ctx,
 
     return result_images;
 }
+
+int sd_get_version(sd_ctx_t* sd_ctx) {
+    if (sd_ctx == NULL) {
+        return VERSION_COUNT;
+    }
+    return sd_ctx->sd->version;
+}
diff --git a/stable-diffusion.h b/stable-diffusion.h
@@ -241,6 +241,8 @@ SD_API sd_image_t* img2vid(sd_ctx_t* sd_ctx,
                            float strength,
                            int64_t seed);
 
+SD_API int sd_get_version(sd_ctx_t* sd_ctx);
+
 typedef struct upscaler_ctx_t upscaler_ctx_t;
 
 SD_API upscaler_ctx_t* new_upscaler_ctx(const char* esrgan_path,
