1 parent bcc7e29 commit 06c359fCopy full SHA for 06c359f
1 file changed
src/model/vae/vae.hpp
@@ -156,7 +156,7 @@ struct VAE : public GGMLRunner {
156
output = _compute(n_threads, input, false);
157
}
158
159
- free_compute_buffer();
+ runner_done();
160
161
if (output.empty()) {
162
LOG_ERROR("vae encode compute failed");
@@ -207,7 +207,7 @@ struct VAE : public GGMLRunner {
207
output = _compute(n_threads, input, true);
208
209
210
211
212
213
LOG_ERROR("vae decode compute failed");
0 commit comments