From 228b80c656d4781a7e6dc2865eaaf955495c3e9a Mon Sep 17 00:00:00 2001 From: qiang Date: Thu, 4 Apr 2024 20:29:35 +0800 Subject: [PATCH] fix tiles_ctx not freed in sd_tiling --- ggml_extend.hpp | 1 + 1 file changed, 1 insertion(+) diff --git a/ggml_extend.hpp b/ggml_extend.hpp index 70e1d17eb..b8008a37c 100644 --- a/ggml_extend.hpp +++ b/ggml_extend.hpp @@ -484,6 +484,7 @@ __STATIC_INLINE__ void sd_tiling(ggml_tensor* input, ggml_tensor* output, const if (tile_count < num_tiles) { pretty_progress(num_tiles, num_tiles, last_time); } + ggml_free(tiles_ctx); } __STATIC_INLINE__ struct ggml_tensor* ggml_group_norm_32(struct ggml_context* ctx,