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;
                     }