Skip to content

Commit 75c26b6

Browse files
committed
refactor: sdxl v-pred
Author: Stéphane du Hamel <stephduh@live.fr>
1 parent 5588fa7 commit 75c26b6

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

stable-diffusion.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -660,6 +660,10 @@ class StableDiffusionGGML {
660660
if (is_using_v_parameterization_for_sd2(ctx, sd_version_is_inpaint(version))) {
661661
is_using_v_parameterization = true;
662662
}
663+
} else if (sd_version_is_sdxl(version)) {
664+
if (model_loader.tensor_storages_types.find("v_pred") != model_loader.tensor_storages_types.end()) {
665+
is_using_v_parameterization = true;
666+
}
663667
} else if (version == VERSION_SVD) {
664668
// TODO: V_PREDICTION_EDM
665669
is_using_v_parameterization = true;

0 commit comments

Comments
 (0)