feat: add sdxl v-pred suppport (leejet#536)

stduhpf · Feb 6, 2025 · 262b9b4 · 262b9b4
1 parent ef7c98a
commit 262b9b4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -527,6 +527,10 @@ class StableDiffusionGGML {
             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;