diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp index b5424ad..cea12e6 100644 --- a/stable-diffusion.cpp +++ b/stable-diffusion.cpp @@ -525,6 +525,10 @@ public: if (is_using_v_parameterization_for_sd2(ctx, sd_version_is_inpaint(version))) { is_using_v_parameterization = true; } + } else if (sd_version_is_sdxl(version)) { + if (model_loader.tensor_storages_types.find("v_pred") != model_loader.tensor_storages_types.end()) { + is_using_v_parameterization = true; + } } else if (version == VERSION_SVD) { // TODO: V_PREDICTION_EDM is_using_v_parameterization = true;