refactor: optimize the handling of sample method (#999)

This commit is contained in:
leejet
2025-11-22 14:00:25 +08:00
committed by GitHub
parent 490c51d963
commit 20345888a3
4 changed files with 64 additions and 52 deletions
+5 -1
View File
@@ -1902,10 +1902,14 @@ int main(int argc, const char* argv[]) {
return 1;
}
if (params.sample_params.sample_method == SAMPLE_METHOD_DEFAULT) {
if (params.sample_params.sample_method == SAMPLE_METHOD_COUNT) {
params.sample_params.sample_method = sd_get_default_sample_method(sd_ctx);
}
if (params.high_noise_sample_params.sample_method == SAMPLE_METHOD_COUNT) {
params.high_noise_sample_params.sample_method = sd_get_default_sample_method(sd_ctx);
}
if (params.sample_params.scheduler == SCHEDULER_COUNT) {
params.sample_params.scheduler = sd_get_default_scheduler(sd_ctx);
}