fix: use sd_get_preview_interval() (#1907)

This commit is contained in:
vmobilis authored and GitHub committed 2026-08-26 00:05:46 +08:00
1 parent 97d2990807
commit 12ee60dc02
1 file changed
+6
+6
View File
@@ -2670,8 +2670,10 @@ public:
denoised = denoised * denoise_mask + init_latent * (1.0f - denoise_mask); denoised = denoised * denoise_mask + init_latent * (1.0f - denoise_mask);
} }
if (sd_should_preview_denoised() && preview.callback != nullptr) { if (sd_should_preview_denoised() && preview.callback != nullptr) {
if (step % sd_get_preview_interval() == 0) {
preview_image(step, denoised, version, preview.mode, preview.callback, preview.data, false); preview_image(step, denoised, version, preview.mode, preview.callback, preview.data, false);
} }
}
report_sample_progress(step, steps, &last_progress_us); report_sample_progress(step, steps, &last_progress_us);
sd::guidance::GuiderOutput output; sd::guidance::GuiderOutput output;
output.pred = denoised; output.pred = denoised;
@@ -2679,8 +2681,10 @@ public:
} }
if (sd_should_preview_noisy() && preview.callback != nullptr) { if (sd_should_preview_noisy() && preview.callback != nullptr) {
if (step % sd_get_preview_interval() == 0) {
preview_image(step, noised_input, version, preview.mode, preview.callback, preview.data, true); preview_image(step, noised_input, version, preview.mode, preview.callback, preview.data, true);
} }
}
sd::Tensor<float> cond_out; sd::Tensor<float> cond_out;
sd::Tensor<float> uncond_out; sd::Tensor<float> uncond_out;
@@ -2889,8 +2893,10 @@ public:
denoised = denoised * denoise_mask + init_latent * (1.0f - denoise_mask); denoised = denoised * denoise_mask + init_latent * (1.0f - denoise_mask);
} }
if (sd_should_preview_denoised() && preview.callback != nullptr) { if (sd_should_preview_denoised() && preview.callback != nullptr) {
if (step % sd_get_preview_interval() == 0) {
preview_image(step, denoised, version, preview.mode, preview.callback, preview.data, false); preview_image(step, denoised, version, preview.mode, preview.callback, preview.data, false);
} }
}
report_sample_progress(step, steps, &last_progress_us); report_sample_progress(step, steps, &last_progress_us);
output.pred = denoised; output.pred = denoised;
return output; return output;