warn if version doesn't support slg

stduhpf · stduhpf · commit bcc888018762 · 2024-11-06T00:14:29.000+01:00
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);
 

Original file line number	Diff line number	Diff line change
`@@ -811,7 +811,12 @@ class StableDiffusionGGML {`
`811`	`811`	`out_uncond = ggml_dup_tensor(tmp_ctx, x);`
`812`	`812`	`}`
`813`	`813`	`if (has_skiplayer) {`
`814`		`- out_skip = ggml_dup_tensor(tmp_ctx, x);`
	`814`	`+ if (version == VERSION_SD3_2B \|\| version == VERSION_SD3_5_2B \|\| version == VERSION_SD3_5_8B \|\| version == VERSION_FLUX_DEV \|\| version == VERSION_FLUX_SCHNELL) {`
	`815`	`+ out_skip = ggml_dup_tensor(tmp_ctx, x);`
	`816`	`+ } else {`
	`817`	`+ has_skiplayer = false;`
	`818`	`+ LOG_WARN("SLG is incompatible with %s models", model_version_to_str[version]);`
	`819`	`+ }`
`815`	`820`	`}`
`816`	`821`	`struct ggml_tensor* denoised = ggml_dup_tensor(tmp_ctx, x);`
`817`	`822`