diff options
| author | Jairo Correa <jn.j41r0@gmail.com> | 2022-09-29 01:14:13 +0000 |
|---|---|---|
| committer | Jairo Correa <jn.j41r0@gmail.com> | 2022-09-29 01:14:13 +0000 |
| commit | c938679de7b87b4f14894d9f57fe0f40dd6e3c06 (patch) | |
| tree | 31119c9dc5d04648bf873069c690b4fd53dc3805 /modules/extras.py | |
| parent | 041d2aefc082c2883aa7e28ee3e4a990b3be9758 (diff) | |
| download | stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.tar.gz stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.tar.bz2 stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.zip | |
Fix memory leak and reduce memory usage
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/extras.py b/modules/extras.py index 9a825530..38b86167 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -98,6 +98,8 @@ def run_extras(extras_mode, image, image_folder, gfpgan_visibility, codeformer_v outputs.append(image)
+ devices.torch_gc()
+
return outputs, plaintext_to_html(info), ''
|