From f6ff06fcb713b531d13ef68a1b8e8a5a1389fe7b Mon Sep 17 00:00:00 2001
From: leejet <leejet714@gmail.com>
Date: Mon, 4 Sep 2023 21:37:46 +0800
Subject: [PATCH] fix: avoid coredump when generating large image

---
 ggml                 | 2 +-
 stable-diffusion.cpp | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/ggml b/ggml
index eec8d3c..6958cd0 160000
--- a/ggml
+++ b/ggml
@@ -1 +1 @@
-Subproject commit eec8d3ca1258d1317a26fdbf561682a7dad4eb01
+Subproject commit 6958cd05c72b2c9461c8b3a270e87beff651797f
diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
index a8e2f98..5d6da17 100644
--- a/stable-diffusion.cpp
+++ b/stable-diffusion.cpp
@@ -3830,6 +3830,7 @@ std::vector<uint8_t> StableDiffusion::txt2img(const std::string& prompt,
     std::vector<uint8_t> result;
     struct ggml_init_params params;
     params.mem_size = static_cast<size_t>(10 * 1024) * 1024;  // 10M
+    params.mem_size += width * height * 3 * sizeof(float) * 2;
     params.mem_buffer = NULL;
     params.no_alloc = false;
     params.dynamic = false;