From d9b5942d988ee36c2f2d8a2d79820e90110947c3 Mon Sep 17 00:00:00 2001
From: stduhpf <stephduh@live.fr>
Date: Sat, 18 Jan 2025 06:15:54 +0100
Subject: [PATCH] feat: add sdxl v-pred suppport (#536)

---
 stable-diffusion.cpp | 4 ++++
 1 file changed, 4 insertions(+)

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;