preview: use new helper functions

stduhpf · thxCode · commit 16d56fb08de8 · 2024-12-13T10:02:14.000+08:00
diff --git a/examples/cli/main.cpp b/examples/cli/main.cpp
@@ -800,9 +800,9 @@ void step_callback(int step, struct ggml_tensor* latents, enum SDVersion version
     if (dim == 16) {
         // 16 channels VAE -> Flux or SD3
 
-        if (version == VERSION_SD3_2B || version == VERSION_SD3_5_8B /* || version == VERSION_SD3_5_2B*/) {
+        if (sd_version_is_sd3(version)) {
             latent_rgb_proj = sd3_latent_rgb_proj;
-        } else if (version == VERSION_FLUX_DEV || version == VERSION_FLUX_SCHNELL) {
+        } else if (sd_version_is_flux(version)) {
             latent_rgb_proj = flux_latent_rgb_proj;
         } else {
             // unknown model
