fix: fix tiles_ctx not freed in sd_tiling (#219)

leejet · Apr 6, 2024 · 646e776 · 646e776
1 parent 3ac48ea
commit 646e776
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,