refactor: sd api

thxCode · thxCode · commit 12f48fd451be · 2024-11-17T19:39:17.000+08:00
Signed-off-by: thxCode &lt;thxcode0824@gmail.com&gt;
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -2104,6 +2104,20 @@ sd_sampling_stream_t* img2img_stream(sd_ctx_t* sd_ctx,
                                control_strength);
 }
 
+int sd_sampling_stream_sampled_steps(sd_sampling_stream_t* stream) {
+    if (stream == nullptr) {
+        return 0;
+    }
+    return stream->sampled_steps;
+}
+
+int sd_sampling_stream_steps(sd_sampling_stream_t* stream) {
+    if (stream == nullptr) {
+        return 0;
+    }
+    return stream->sample_steps + 1;
+}
+
 void sd_sampling_stream_free(sd_sampling_stream_t* stream) {
     if (stream == nullptr) {
         return;
diff --git a/stable-diffusion.h b/stable-diffusion.h
@@ -258,6 +258,8 @@ SD_API sd_sampling_stream_t* img2img_stream(sd_ctx_t* sd_ctx,
                                             int64_t seed,
                                             const sd_image_t* control_cond,
                                             float control_strength);
+SD_API int sd_sampling_stream_sampled_steps(sd_sampling_stream_t* stream);
+SD_API int sd_sampling_stream_steps(sd_sampling_stream_t* stream);
 SD_API void sd_sampling_stream_free(sd_sampling_stream_t* stream);
 SD_API bool sd_sampling_stream_sample(sd_ctx_t* sd_ctx, sd_sampling_stream_t* stream);
 SD_API sd_image_t sd_samping_stream_get_image(sd_ctx_t* sd_ctx, sd_sampling_stream_t* stream);
