Increase context memory when previewing with auto encoder instead

leejet · Dec 14, 2024 · e673fb5 · e673fb5
1 parent 2a23987
commit e673fb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -1611,7 +1611,7 @@ sd_image_t* txt2img(sd_ctx_t* sd_ctx,
     if (sd_ctx->sd->stacked_id) {
         params.mem_size += static_cast<size_t>(10 * 1024 * 1024);  // 10 MB
     }
-    if (sd_ctx->sd->first_stage_model != nullptr && sd_ctx->sd->tae_first_stage != nullptr) {
+    if (preview_mode!=SD_PREVIEW_NONE && preview_mode!=SD_PREVIEW_PROJ) {
         params.mem_size *= 2;
     }
     params.mem_size += width * height * 3 * sizeof(float);