From 428f5ee00e7d64e73ae79a853a3041c28d1a5155 Mon Sep 17 00:00:00 2001 From: Green Sky Date: Fri, 17 Apr 2026 11:22:03 +0200 Subject: [PATCH] fix: ernie-image flow shift see: - https://huggingface.co/baidu/ERNIE-Image-Turbo/blob/main/scheduler/scheduler_config.json - https://huggingface.co/baidu/ERNIE-Image/blob/main/scheduler/scheduler_config.json - https://huggingface.co/baidu/ERNIE-Image/discussions/3 --- src/stable-diffusion.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/stable-diffusion.cpp b/src/stable-diffusion.cpp index 42ed2d46e..af8bd7001 100644 --- a/src/stable-diffusion.cpp +++ b/src/stable-diffusion.cpp @@ -943,6 +943,8 @@ class StableDiffusionGGML { pred_type = FLOW_PRED; if (sd_version_is_wan(version)) { default_flow_shift = 5.f; + } else if (sd_version_is_ernie_image(version)) { + default_flow_shift = 4.f; } else { default_flow_shift = 3.f; }