alphard/stable-diffusion.cpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: avoid potential dangling pointer problem

Browse Source
This commit is contained in:
leejet 2025-03-01 16:58:26 +08:00
parent 195d170136
commit 30b3ac8e62
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/cli/main.cpp
View File

@ -931,12 +931,12 @@ int main(int argc, const char* argv[]) {
} }
} }
std::vector<uint8_t> default_mask_image_vec(params.width * params.height, 255);
if (params.mask_path != "") { if (params.mask_path != "") {
int c = 0; int c = 0;
mask_image_buffer = stbi_load(params.mask_path.c_str(), &params.width, &params.height, &c, 1); mask_image_buffer = stbi_load(params.mask_path.c_str(), &params.width, &params.height, &c, 1);
} else { } else {
std::vector<uint8_t> arr(params.width * params.height, 255); mask_image_buffer = default_mask_image_vec.data();
mask_image_buffer = arr.data();
} }
sd_image_t mask_image = {(uint32_t)params.width, sd_image_t mask_image = {(uint32_t)params.width,
(uint32_t)params.height, (uint32_t)params.height,