upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-07-30 06:40:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

perf: free unused params immediately to reduce memory usage

Browse Source
This commit is contained in:
leejet
2023-08-17 00:55:36 +08:00
parent cbee3c9a4f
commit 8f34dd7cc7
4 changed files with 232 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
main.cpp
View File

@@ -322,7 +322,8 @@ int main(int argc, const char* argv[]) {
}
init_img.assign(img_data, img_data + (opt.w * opt.h * c));
}
StableDiffusion sd(opt.n_threads, vae_decode_only);
StableDiffusion sd(opt.n_threads, vae_decode_only, true);
if (!sd.load_from_file(opt.model_path)) {
return 1;
}