DarthAffe/stable-diffusion.cpp
1
0 Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-11-11 11:26:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct step count adjustment for custom sigmas list (#2084)

Browse Source
This commit is contained in:
Wagner Bruna authored and GitHub committed 2026-10-06 17:55:00 +08:00
1 parent f4b5326da7
commit c150a6bafe
1 file changed
+4 -2
Download Patch File Download Diff File Expand all files Collapse all files
src/pipeline/request.cpp
+4 -2
View File
@@ -377,9 +377,10 @@ namespace sd::pipeline {
if (sample_params->custom_sigmas_count > 0) {
sigmas = std::vector<float>(sample_params->custom_sigmas,
sample_params->custom_sigmas + sample_params->custom_sigmas_count);
total_steps = static_cast<int>(sigmas.size()) - 1;
if (total_steps != sample_params->custom_sigmas_count - 1) {
total_steps = sample_params->custom_sigmas_count - 1;
LOG_WARN("total_steps != custom_sigmas_count - 1, set total_steps to %d", total_steps);
if (sample_steps >= total_steps) {
if ((sample_steps > total_steps) || (high_noise_sample_steps <= 0 && sample_steps != total_steps)) {
sample_steps = total_steps;
LOG_WARN("total_steps != custom_sigmas_count - 1, set sample_steps to %d", sample_steps);
}
@@ -387,6 +388,7 @@ namespace sd::pipeline {
high_noise_sample_steps = total_steps - sample_steps;
LOG_WARN("total_steps != custom_sigmas_count - 1, set high_noise_sample_steps to %d", high_noise_sample_steps);
}
}
} else {
scheduler_t scheduler = resolve_scheduler(sd,
sample_params->scheduler,