Skip to content

Commit 06c359f

Browse files
authored
fix: replace free_compute_buffer with runner_done in vae (#1872)
1 parent bcc7e29 commit 06c359f

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/model/vae/vae.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ struct VAE : public GGMLRunner {
156156
output = _compute(n_threads, input, false);
157157
}
158158

159-
free_compute_buffer();
159+
runner_done();
160160

161161
if (output.empty()) {
162162
LOG_ERROR("vae encode compute failed");
@@ -207,7 +207,7 @@ struct VAE : public GGMLRunner {
207207
output = _compute(n_threads, input, true);
208208
}
209209

210-
free_compute_buffer();
210+
runner_done();
211211

212212
if (output.empty()) {
213213
LOG_ERROR("vae decode compute failed");

0 commit comments

Comments
 (0)