upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-07-30 14:50:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: add support for custom scheduler (#694)

Browse Source 
---------

Co-authored-by: leejet <leejet714@gmail.com>
This commit is contained in:
rmatif
2025-12-13 09:20:02 +01:00
committed by GitHub
parent 15d0f82760
commit 8f05f5bc6e
6 changed files with 101 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/cli/main.cpp
View File

@@ -258,7 +258,15 @@ std::string get_image_params(const SDCliParams& cli_params, const SDContextParam
parameter_string += "Sampler RNG: " + std::string(sd_rng_type_name(ctx_params.sampler_rng_type)) + ", ";
}
parameter_string += "Sampler: " + std::string(sd_sample_method_name(gen_params.sample_params.sample_method));
if (gen_params.sample_params.scheduler != SCHEDULER_COUNT) {
if (!gen_params.custom_sigmas.empty()) {
parameter_string += ", Custom Sigmas: [";
for (size_t i = 0; i < gen_params.custom_sigmas.size(); ++i) {
std::ostringstream oss;
oss << std::fixed << std::setprecision(4) << gen_params.custom_sigmas[i];
parameter_string += oss.str() + (i == gen_params.custom_sigmas.size() - 1 ? "" : ", ");
}
parameter_string += "]";
} else if (gen_params.sample_params.scheduler != SCHEDULER_COUNT) { // Only show schedule if not using custom sigmas
parameter_string += " " + std::string(sd_scheduler_name(gen_params.sample_params.scheduler));
}
parameter_string += ", ";
@@ -806,4 +814,4 @@ int main(int argc, const char* argv[]) {
release_all_resources();
return 0;
}
}