Increase context memory when previewing with auto encoder instead

stduhpf · stduhpf · commit e673fb5a8675 · 2024-12-14T02:13:16.000+01:00
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);

Original file line number	Diff line number	Diff line change
`@@ -1611,7 +1611,7 @@ sd_image_t* txt2img(sd_ctx_t* sd_ctx,`
`1611`	`1611`	`if (sd_ctx->sd->stacked_id) {`
`1612`	`1612`	`params.mem_size += static_cast<size_t>(10 * 1024 * 1024); // 10 MB`
`1613`	`1613`	`}`
`1614`		`- if (sd_ctx->sd->first_stage_model != nullptr && sd_ctx->sd->tae_first_stage != nullptr) {`
	`1614`	`+ if (preview_mode!=SD_PREVIEW_NONE && preview_mode!=SD_PREVIEW_PROJ) {`
`1615`	`1615`	`params.mem_size *= 2;`
`1616`	`1616`	`}`
`1617`	`1617`	`params.mem_size += width * height * 3 * sizeof(float);`