warn if version doesn't support slg

stduhpf · Nov 5, 2024 · bcc8880 · bcc8880
1 parent 4fe0e82
commit bcc8880
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -811,7 +811,12 @@ class StableDiffusionGGML {
             out_uncond = ggml_dup_tensor(tmp_ctx, x);
         }
         if (has_skiplayer) {
-            out_skip = ggml_dup_tensor(tmp_ctx, x);
+            if (version == VERSION_SD3_2B || version == VERSION_SD3_5_2B || version == VERSION_SD3_5_8B || version == VERSION_FLUX_DEV || version == VERSION_FLUX_SCHNELL) {
+                out_skip = ggml_dup_tensor(tmp_ctx, x);
+            } else {
+                has_skiplayer = false;
+                LOG_WARN("SLG is incompatible with %s models", model_version_to_str[version]);
+            }
         }
         struct ggml_tensor* denoised = ggml_dup_tensor(tmp_ctx, x);