Skip to content

Commit d9b5942

Browse files
authored
feat: add sdxl v-pred suppport (#536)
1 parent 587a37b commit d9b5942

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
@@ -525,6 +525,10 @@ class StableDiffusionGGML {
525525
if (is_using_v_parameterization_for_sd2(ctx, sd_version_is_inpaint(version))) {
526526
is_using_v_parameterization = true;
527527
}
528+
} else if (sd_version_is_sdxl(version)) {
529+
if (model_loader.tensor_storages_types.find("v_pred") != model_loader.tensor_storages_types.end()) {
530+
is_using_v_parameterization = true;
531+
}
528532
} else if (version == VERSION_SVD) {
529533
// TODO: V_PREDICTION_EDM
530534
is_using_v_parameterization = true;

0 commit comments

Comments
 (0)