Skip to content

Commit 262b9b4

Browse files
committed
feat: add sdxl v-pred suppport (leejet#536)
1 parent ef7c98a commit 262b9b4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

stable-diffusion.cpp

+4
Original file line numberDiff line numberDiff line change
@@ -527,6 +527,10 @@ class StableDiffusionGGML {
527527
if (is_using_v_parameterization_for_sd2(ctx, sd_version_is_inpaint(version))) {
528528
is_using_v_parameterization = true;
529529
}
530+
} else if (sd_version_is_sdxl(version)) {
531+
if (model_loader.tensor_storages_types.find("v_pred") != model_loader.tensor_storages_types.end()) {
532+
is_using_v_parameterization = true;
533+
}
530534
} else if (version == VERSION_SVD) {
531535
// TODO: V_PREDICTION_EDM
532536
is_using_v_parameterization = true;

0 commit comments

Comments
 (0)