fix(tx): align ggml_type to sd_type

thxCode · thxCode · commit 4fb14fe9535f · 2024-12-02T14:46:31.000+08:00
Signed-off-by: thxCode &lt;thxcode0824@gmail.com&gt;
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -322,7 +322,7 @@ class StableDiffusionGGML {
         ggml_type t5xxl_wtype           = GGML_TYPE_COUNT;
         ggml_type diffusion_model_wtype = GGML_TYPE_COUNT;
         ggml_type vae_wtype             = GGML_TYPE_COUNT;
-        if (wtype == GGML_TYPE_COUNT) {
+        if (wtype >= GGML_TYPE_COUNT) {
             model_wtype = model_loader.get_sd_wtype();
             if (model_wtype == GGML_TYPE_COUNT) {
                 model_wtype = GGML_TYPE_F32;
