fix preview with unet inpaint models

Author: stduhpf

stable-diffusion.cpp

@@ -866,9 +866,9 @@ void preview_image(ggml_context* work_ctx,
 
             } else if (dim == 4) {
                 // 4 channels VAE
-                if (version == VERSION_SDXL) {
+                if (sd_version_is_sdxl(version)) {
                     latent_rgb_proj = sdxl_latent_rgb_proj;
-                } else if (version == VERSION_SD1 || version == VERSION_SD2) {
+                } else if (sd_version_is_sd1(version) || sd_version_is_sd2(version)) {
                     latent_rgb_proj = sd_latent_rgb_proj;
                 } else {
                     // unknown model
@@ -2017,7 +2017,7 @@ sd_image_t* generate_image(sd_ctx_t* sd_ctx, const sd_img_gen_params_t* sd_img_g
     if (sd_ctx->sd->stacked_id) {
         params.mem_size += static_cast<size_t>(10 * 1024 * 1024);  // 10 MB
     }
-    if (preview_mode!=SD_PREVIEW_NONE && preview_mode!=SD_PREVIEW_PROJ) {
+    if (preview_mode != SD_PREVIEW_NONE && preview_mode != SD_PREVIEW_PROJ) {
         params.mem_size *= 2;
     }
     params.mem_size += width * height * 3 * sizeof(float) * 3;