mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-10-03 03:17:51 -05:00
feat: turn flow_shift into a generation parameter (#1289)
* feat: turn flow_shift into a generation parameter * format code * simplify set_shift/set_parameters * fix sd_sample_params_to_str * remove unused variable * update docs --------- Co-authored-by: leejet <leejet714@gmail.com>
This commit is contained in:
@@ -581,10 +581,6 @@ struct SDContextParams {
|
||||
"--vae-tile-overlap",
|
||||
"tile overlap for vae tiling, in fraction of tile size (default: 0.5)",
|
||||
&vae_tiling_params.target_overlap},
|
||||
{"",
|
||||
"--flow-shift",
|
||||
"shift value for Flow models like SD3.x or WAN (default: auto)",
|
||||
&flow_shift},
|
||||
};
|
||||
|
||||
options.bool_options = {
|
||||
@@ -903,7 +899,6 @@ struct SDContextParams {
|
||||
<< " photo_maker_path: \"" << photo_maker_path << "\",\n"
|
||||
<< " rng_type: " << sd_rng_type_name(rng_type) << ",\n"
|
||||
<< " sampler_rng_type: " << sd_rng_type_name(sampler_rng_type) << ",\n"
|
||||
<< " flow_shift: " << (std::isinf(flow_shift) ? "INF" : std::to_string(flow_shift)) << "\n"
|
||||
<< " offload_params_to_cpu: " << (offload_params_to_cpu ? "true" : "false") << ",\n"
|
||||
<< " enable_mmap: " << (enable_mmap ? "true" : "false") << ",\n"
|
||||
<< " control_net_cpu: " << (control_net_cpu ? "true" : "false") << ",\n"
|
||||
@@ -986,7 +981,6 @@ struct SDContextParams {
|
||||
chroma_use_t5_mask,
|
||||
chroma_t5_mask_pad,
|
||||
qwen_image_zero_cond_t,
|
||||
flow_shift,
|
||||
};
|
||||
return sd_ctx_params;
|
||||
}
|
||||
@@ -1206,6 +1200,10 @@ struct SDGenerationParams {
|
||||
"--eta",
|
||||
"eta in DDIM, only for DDIM and TCD (default: 0)",
|
||||
&sample_params.eta},
|
||||
{"",
|
||||
"--flow-shift",
|
||||
"shift value for Flow models like SD3.x or WAN (default: auto)",
|
||||
&sample_params.flow_shift},
|
||||
{"",
|
||||
"--high-noise-cfg-scale",
|
||||
"(high noise) unconditional guidance scale: (default: 7.0)",
|
||||
@@ -1606,6 +1604,7 @@ struct SDGenerationParams {
|
||||
load_if_exists("cfg_scale", sample_params.guidance.txt_cfg);
|
||||
load_if_exists("img_cfg_scale", sample_params.guidance.img_cfg);
|
||||
load_if_exists("guidance", sample_params.guidance.distilled_guidance);
|
||||
load_if_exists("flow_shift", sample_params.flow_shift);
|
||||
|
||||
auto load_sampler_if_exists = [&](const char* key, enum sample_method_t& out) {
|
||||
if (j.contains(key) && j[key].is_string()) {
|
||||
|
||||
Reference in New Issue
Block a user