Skip to content

Commit 6a9cb31

Browse files
authored
fix: tune ernie-image default flow shift (#1433)
1 parent 2bcff67 commit 6a9cb31

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/stable-diffusion.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -943,6 +943,8 @@ class StableDiffusionGGML {
943943
pred_type = FLOW_PRED;
944944
if (sd_version_is_wan(version)) {
945945
default_flow_shift = 5.f;
946+
} else if (sd_version_is_ernie_image(version)) {
947+
default_flow_shift = 4.f;
946948
} else {
947949
default_flow_shift = 3.f;
948950
}

0 commit comments

Comments
 (0)