fix: avoid some memory leaks (#136)
--------- Co-authored-by: leejet <leejet714@gmail.com>
This commit is contained in:
parent
4a5e7b58e2
commit
f2e4d9793b
@ -560,6 +560,7 @@ int main(int argc, const char* argv[]) {
|
||||
|
||||
if (results == NULL) {
|
||||
printf("generate failed\n");
|
||||
free_sd_ctx(sd_ctx);
|
||||
return 1;
|
||||
}
|
||||
|
||||
@ -600,6 +601,8 @@ int main(int argc, const char* argv[]) {
|
||||
free(results[i].data);
|
||||
results[i].data = NULL;
|
||||
}
|
||||
free(results);
|
||||
free_sd_ctx(sd_ctx);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
@ -101,6 +101,7 @@ public:
|
||||
}
|
||||
|
||||
~StableDiffusionGGML() {
|
||||
ggml_backend_free(backend);
|
||||
}
|
||||
|
||||
bool load_from_file(const std::string& model_path,
|
||||
|
Loading…
Reference in New Issue
Block a user