Skip to content

Commit b8f7337

Browse files
committed
Fix server
1 parent ed45878 commit b8f7337

File tree

1 file changed

+15
-12
lines changed

1 file changed

+15
-12
lines changed

examples/server/main.cpp

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1484,14 +1484,26 @@ void start_server(SDParams params) {
14841484
}
14851485

14861486
{
1487+
sd_guidance_params_t guidance_params = {params.lastRequest.cfg_scale,
1488+
params.lastRequest.cfg_scale,
1489+
params.lastRequest.min_cfg,
1490+
params.lastRequest.guidance,
1491+
{params.lastRequest.skip_layers.data(),
1492+
params.lastRequest.skip_layers.size(),
1493+
params.lastRequest.slg_scale,
1494+
params.lastRequest.skip_layer_start,
1495+
params.lastRequest.skip_layer_end},
1496+
{params.lastRequest.apg_eta,
1497+
params.lastRequest.apg_momentum,
1498+
params.lastRequest.apg_norm_threshold,
1499+
params.lastRequest.apg_norm_smoothing}};
14871500
sd_set_preview_callback((sd_preview_cb_t)step_callback, params.lastRequest.preview_method, params.lastRequest.preview_interval);
14881501
sd_image_t* results;
14891502
results = txt2img(sd_ctx,
14901503
params.lastRequest.prompt.c_str(),
14911504
params.lastRequest.negative_prompt.c_str(),
14921505
params.lastRequest.clip_skip,
1493-
params.lastRequest.cfg_scale,
1494-
params.lastRequest.guidance,
1506+
guidance_params,
14951507
0.,
14961508
params.lastRequest.width,
14971509
params.lastRequest.height,
@@ -1503,16 +1515,7 @@ void start_server(SDParams params) {
15031515
1,
15041516
params.lastRequest.style_ratio,
15051517
params.lastRequest.normalize_input,
1506-
params.input_id_images_path.c_str(),
1507-
sd_slg_params_t{params.lastRequest.skip_layers.data(),
1508-
params.lastRequest.skip_layers.size(),
1509-
params.lastRequest.slg_scale,
1510-
params.lastRequest.skip_layer_start,
1511-
params.lastRequest.skip_layer_end},
1512-
sd_apg_params_t{params.lastRequest.apg_eta,
1513-
params.lastRequest.apg_momentum,
1514-
params.lastRequest.apg_norm_threshold,
1515-
params.lastRequest.apg_norm_smoothing});
1518+
params.input_id_images_path.c_str());
15161519

15171520
if (results == NULL) {
15181521
printf("generate failed\n");

0 commit comments

Comments
 (0)