Increase context memory when previewing with auto encoder instead

stduhpf · stduhpf · commit 306005368963 · 2025-02-06T16:03:30.000+01:00
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
@@ -1676,7 +1676,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
`@@ -1676,7 +1676,7 @@ sd_image_t* txt2img(sd_ctx_t* sd_ctx,`
`1676`	`1676`	`if (sd_ctx->sd->stacked_id) {`
`1677`	`1677`	`params.mem_size += static_cast<size_t>(10 * 1024 * 1024); // 10 MB`
`1678`	`1678`	`}`
`1679`		`- if (sd_ctx->sd->first_stage_model != nullptr && sd_ctx->sd->tae_first_stage != nullptr) {`
	`1679`	`+ if (preview_mode!=SD_PREVIEW_NONE && preview_mode!=SD_PREVIEW_PROJ) {`
`1680`	`1680`	`params.mem_size *= 2;`
`1681`	`1681`	`}`
`1682`	`1682`	`params.mem_size += width * height * 3 * sizeof(float);`