Skip to content

Commit 03d9612

Browse files
committed
server: rebase on lastest master
1 parent 0079543 commit 03d9612

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

examples/server/main.cpp

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -631,7 +631,7 @@ void parse_args(int argc, const char** argv, SDParams& params) {
631631
}
632632

633633
if (params.ctxParams.n_threads <= 0) {
634-
params.ctxParams.n_threads = get_num_physical_cores();
634+
params.ctxParams.n_threads = sd_get_num_physical_cores();
635635
}
636636

637637
if (params.lastRequest.prompt.length() == 0) {
@@ -678,7 +678,7 @@ void parse_args(int argc, const char** argv, SDParams& params) {
678678
}
679679

680680
if (params.ctxParams.n_threads <= 0) {
681-
params.ctxParams.n_threads = get_num_physical_cores();
681+
params.ctxParams.n_threads = sd_get_num_physical_cores();
682682
}
683683
}
684684

@@ -1836,7 +1836,8 @@ void add_task(std::string task_id, std::function<void()> task) {
18361836
const char* preview_path;
18371837
float preview_fps = 24; // TODO : video
18381838

1839-
void step_callback(int step, int frame_count, sd_image_t* image, bool is_noisy) {
1839+
void step_callback(int step, int frame_count, sd_image_t* image, bool is_noisy, void* data) {
1840+
(void) data;
18401841
using json = nlohmann::json;
18411842
if (frame_count > 1) {
18421843
return;
@@ -2059,7 +2060,7 @@ nlohmann::json serv_generate_image(sd_ctx_t*& sd_ctx, SDParams& params, int& n_p
20592060
params.lastRequest.control_strength,
20602061
params.lastRequest.pm_params,
20612062
params.lastRequest.tiling_params};
2062-
sd_set_preview_callback((sd_preview_cb_t)step_callback, params.lastRequest.preview_method, params.lastRequest.preview_interval, !params.lastRequest.preview_noisy, params.lastRequest.preview_noisy);
2063+
sd_set_preview_callback((sd_preview_cb_t)step_callback, params.lastRequest.preview_method, params.lastRequest.preview_interval, !params.lastRequest.preview_noisy, params.lastRequest.preview_noisy, NULL);
20632064

20642065
results = generate_image(sd_ctx, &gen_params);
20652066

0 commit comments

Comments
 (0)