diff options
| author | Jairo Correa <jn.j41r0@gmail.com> | 2022-11-02 00:56:47 +0000 |
|---|---|---|
| committer | Jairo Correa <jn.j41r0@gmail.com> | 2022-11-02 00:56:47 +0000 |
| commit | c9148b2312b36fee8727f5233da9dbe32aa1f58c (patch) | |
| tree | 01440de3870752ef7e5e2ebef5dfa894622684ac /modules/processing.py | |
| parent | 61af311f29d0fde6db7027cf8aa68f0ae9f6455f (diff) | |
| download | stable-diffusion-webui-gfx803-c9148b2312b36fee8727f5233da9dbe32aa1f58c.tar.gz stable-diffusion-webui-gfx803-c9148b2312b36fee8727f5233da9dbe32aa1f58c.tar.bz2 stable-diffusion-webui-gfx803-c9148b2312b36fee8727f5233da9dbe32aa1f58c.zip | |
Release processing resources after it finishes
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/modules/processing.py b/modules/processing.py index 57d3a523..b541ee2b 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -202,6 +202,10 @@ class StableDiffusionProcessing(): def sample(self, conditioning, unconditional_conditioning, seeds, subseeds, subseed_strength):
raise NotImplementedError()
+ def close(self):
+ self.sd_model = None
+ self.sampler = None
+
class Processed:
def __init__(self, p: StableDiffusionProcessing, images_list, seed=-1, info="", subseed=None, all_prompts=None, all_seeds=None, all_subseeds=None, index_of_first_image=0, infotexts=None):
@@ -597,9 +601,6 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: if p.scripts is not None:
p.scripts.postprocess(p, res)
- p.sd_model = None
- p.sampler = None
-
return res
|