From d637cfb935feb2254898a8f1524046c958f3df80 Mon Sep 17 00:00:00 2001 From: leejet Date: Mon, 12 Oct 2026 23:09:42 +0800 Subject: [PATCH] feat: add Qwen Image 2.1 Turbo sigma preset (#2130) --- docs/qwen_image_2.1.md | 10 ++++++++++ examples/common/common.cpp | 7 +++++-- src/pipeline/diffusion_engine.cpp | 6 +++--- 3 files changed, 18 insertions(+), 5 deletions(-) diff --git a/docs/qwen_image_2.1.md b/docs/qwen_image_2.1.md index 5da89c53..475404c7 100644 --- a/docs/qwen_image_2.1.md +++ b/docs/qwen_image_2.1.md @@ -32,6 +32,16 @@ Run the following commands from the build directory. Use image dimensions divisi To use GGUF diffusion weights, set `--diffusion-model` to the path of a file such as `qwen_image_2.1-Q4_K.gguf`. +### Turbo + +For [Qwen Image 2.1 Turbo](https://huggingface.co/Qwen/Qwen-Image-2.1-Turbo), use the Turbo diffusion weights and select its fixed 8-step sigma schedule with `--sigmas qwen-image-2.1-turbo`. For example, add these sampling options to the command above: + +```text +--sigmas qwen-image-2.1-turbo --steps 8 --cfg-scale 1.0 --sampling-method euler +``` + +The preset is equivalent to `--sigmas 1.0,0.978453,0.95418,0.926626,0.89508,0.845148,0.704534,0.414568,0.0`. It only supplies the sigma values; CFG scale and sampling method remain independently configurable. As with an explicit sigma list, the sigma count determines the actual number of sampling steps. The same preset name is also accepted by `--hires-sigmas` for the highres fix second pass. + ### Image editing Pass the reference image with `-r` and describe the edit in `-p`. Vision weights are required; the example below loads them separately with `--llm_vision`. diff --git a/examples/common/common.cpp b/examples/common/common.cpp index faa63726..46d33ce1 100644 --- a/examples/common/common.cpp +++ b/examples/common/common.cpp @@ -1475,6 +1475,9 @@ ArgOptions SDGenerationParams::get_options() { return -1; } std::string sigmas_str = value; + if (sigmas_str == "qwen-image-2.1-turbo") { + sigmas_str = "1.0,0.978453,0.95418,0.926626,0.89508,0.845148,0.704534,0.414568,0.0"; + } if (!sigmas_str.empty() && sigmas_str.front() == '[') { sigmas_str.erase(0, 1); } @@ -1721,11 +1724,11 @@ ArgOptions SDGenerationParams::get_options() { on_scheduler_arg}, {"", "--sigmas", - "custom sigma values for the sampler, comma-separated (e.g., \"14.61,7.8,3.5,0.0\").", + "custom sigma values for the sampler, comma-separated (e.g., \"14.61,7.8,3.5,0.0\") or preset: qwen-image-2.1-turbo.", on_sigmas_arg}, {"", "--hires-sigmas", - "custom sigma values for the highres fix second pass, comma-separated (e.g., \"0.85,0.725,0.421875,0.0\").", + "custom sigma values for the highres fix second pass, comma-separated (e.g., \"0.85,0.725,0.421875,0.0\") or preset: qwen-image-2.1-turbo.", on_hires_sigmas_arg}, {"", "--skip-layers", diff --git a/src/pipeline/diffusion_engine.cpp b/src/pipeline/diffusion_engine.cpp index 6400b243..ddc4eb15 100644 --- a/src/pipeline/diffusion_engine.cpp +++ b/src/pipeline/diffusion_engine.cpp @@ -2735,9 +2735,9 @@ sd::Tensor StableDiffusionGGML::sample(const std::shared_ptr batched_out = run_batched_condition(*positive_condition, c_concat_override); if (!batched_out.empty() && batched_out.dim() >= 4 && batched_out.shape()[3] == 2) { - auto parts = sd::ops::chunk(batched_out, 2, 3); - cond_out = std::move(parts[0]); - uncond_out = std::move(parts[1]); + auto parts = sd::ops::chunk(batched_out, 2, 3); + cond_out = std::move(parts[0]); + uncond_out = std::move(parts[1]); } }