Skip to content

Commit 9602849

Browse files
committed
fix img2img channels
1 parent 060b579 commit 9602849

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/server/main.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1232,7 +1232,7 @@ bool parseJsonPrompt(std::string json_str, SDParams* params) {
12321232
if (params->lastRequest.init_image.data) {
12331233
free(params->lastRequest.init_image.data);
12341234
}
1235-
params->lastRequest.init_image = sd_image_t{(uint32_t)width, (uint32_t)height, (uint32_t)c, image_buffer};
1235+
params->lastRequest.init_image = sd_image_t{(uint32_t)width, (uint32_t)height, 3, image_buffer};
12361236
sd_log(sd_log_level_t::SD_LOG_INFO, "Loaded image from memory: %dx%d, %d channels\n", width, height, c);
12371237
return true;
12381238
}},
@@ -1312,7 +1312,7 @@ bool parseJsonPrompt(std::string json_str, SDParams* params) {
13121312
if (image_buffer == NULL) {
13131313
sd_log(sd_log_level_t::SD_LOG_WARN, "Failed to load image from memory\n");
13141314
}
1315-
params->lastRequest.ref_images.push_back(sd_image_t{(uint32_t)width, (uint32_t)height, (uint32_t)c, image_buffer});
1315+
params->lastRequest.ref_images.push_back(sd_image_t{(uint32_t)width, (uint32_t)height, 3, image_buffer});
13161316
}
13171317
return true;
13181318
}},

0 commit comments

Comments
 (0)