DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2025-12-12 13:28:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: avoid crash on default video preview path (#997)

Browse Source 
Co-authored-by: masamaru-san
This commit is contained in:
Wagner Bruna 2025-11-22 01:46:27 -03:00 committed by GitHub
parent 5498cc0d67
commit b542894fb9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
examples/cli/main.cpp
View File

@ -1648,7 +1648,7 @@ bool load_images_from_dir(const std::string dir,
return true; return true;
} }
const char* preview_path; std::string preview_path;
float preview_fps; float preview_fps;
void step_callback(int step, int frame_count, sd_image_t* image, bool is_noisy) { void step_callback(int step, int frame_count, sd_image_t* image, bool is_noisy) {
@ -1657,16 +1657,16 @@ void step_callback(int step, int frame_count, sd_image_t* image, bool is_noisy)
// is_noisy is set to true if the preview corresponds to noisy latents, false if it's denoised latents // is_noisy is set to true if the preview corresponds to noisy latents, false if it's denoised latents
// unused in this app, it will either be always noisy or always denoised here // unused in this app, it will either be always noisy or always denoised here
if (frame_count == 1) { if (frame_count == 1) {
stbi_write_png(preview_path, image->width, image->height, image->channel, image->data, 0); stbi_write_png(preview_path.c_str(), image->width, image->height, image->channel, image->data, 0);
} else { } else {
create_mjpg_avi_from_sd_images(preview_path, image, frame_count, preview_fps); create_mjpg_avi_from_sd_images(preview_path.c_str(), image, frame_count, preview_fps);
} }
} }
int main(int argc, const char* argv[]) { int main(int argc, const char* argv[]) {
SDParams params; SDParams params;
parse_args(argc, argv, params); parse_args(argc, argv, params);
preview_path = params.preview_path.c_str(); preview_path = params.preview_path;
if (params.video_frames > 4) { if (params.video_frames > 4) {
size_t last_dot_pos = params.preview_path.find_last_of("."); size_t last_dot_pos = params.preview_path.find_last_of(".");
std::string base_path = params.preview_path; std::string base_path = params.preview_path;
@ -1677,8 +1677,7 @@ int main(int argc, const char* argv[]) {
std::transform(file_ext.begin(), file_ext.end(), file_ext.begin(), ::tolower); std::transform(file_ext.begin(), file_ext.end(), file_ext.begin(), ::tolower);
} }
if (file_ext == ".png") { if (file_ext == ".png") {
base_path = base_path + ".avi"; preview_path = base_path + ".avi";
preview_path = base_path.c_str();
} }
} }
preview_fps = params.fps; preview_fps = params.fps;