refactor(tx): add default params util

thxCode · thxCode · commit e660807cc985 · 2024-11-27T22:05:40.000+08:00
Signed-off-by: thxCode &lt;thxcode0824@gmail.com&gt;
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -1846,3 +1846,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
@@ -209,6 +209,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,
