Increase context memory when previewing with auto encoder instead

stduhpf · Stéphane du Hamel · commit 1d38c49cdd1c · 2024-12-28T12:19:58.000+01:00
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -1670,7 +1670,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);

Original file line number	Diff line number	Diff line change
`@@ -1670,7 +1670,7 @@ sd_image_t* txt2img(sd_ctx_t* sd_ctx,`
`1670`	`1670`	`if (sd_ctx->sd->stacked_id) {`
`1671`	`1671`	`params.mem_size += static_cast<size_t>(10 * 1024 * 1024); // 10 MB`
`1672`	`1672`	`}`
`1673`		`- if (sd_ctx->sd->first_stage_model != nullptr && sd_ctx->sd->tae_first_stage != nullptr) {`
	`1673`	`+ if (preview_mode!=SD_PREVIEW_NONE && preview_mode!=SD_PREVIEW_PROJ) {`
`1674`	`1674`	`params.mem_size *= 2;`
`1675`	`1675`	`}`
`1676`	`1676`	`params.mem_size += width * height * 3 * sizeof(float);`