lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-11 07:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

make clear_prompt_cache a function

Browse Source
This commit is contained in:
layerdiffusion
2024-08-17 19:00:13 -07:00
parent 4bb5613916
commit 0f266c48bd
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
modules/processing.py
View File

@@ -212,6 +212,12 @@ class StableDiffusionProcessing:
latents_after_sampling = []
pixels_after_sampling = []
def clear_prompt_cache(self):
self.cached_c = [None, None, None]
self.cached_uc = [None, None, None]
StableDiffusionProcessing.cached_c = [None, None, None]
StableDiffusionProcessing.cached_uc = [None, None, None]
def __post_init__(self):
if self.sampler_index is not None:
print("sampler_index argument for StableDiffusionProcessing does not do anything; use sampler_name", file=sys.stderr)
@@ -792,10 +798,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
memory_management.unload_all_models()
if need_global_unload:
StableDiffusionProcessing.cached_c = [None, None, None]
StableDiffusionProcessing.cached_uc = [None, None, None]
p.cached_c = [None, None, None]
p.cached_uc = [None, None, None]
p.clear_prompt_cache()
need_global_unload = False