From 4d0ba8692a0e829cb0a80aab9211b467be7d7490 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sun, 23 Jun 2024 04:50:30 +0300 Subject: [PATCH] Add gc to prompt parser --- modules/prompt_parser_diffusers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/prompt_parser_diffusers.py b/modules/prompt_parser_diffusers.py index 18dadc408..3d1ef1f5b 100644 --- a/modules/prompt_parser_diffusers.py +++ b/modules/prompt_parser_diffusers.py @@ -199,6 +199,7 @@ def encode_prompts(pipe, p, prompts: list, negative_prompts: list, steps: int, c if debug_enabled: get_tokens('positive', prompts[0]) get_tokens('negative', negative_prompts[0]) + devices.torch_gc() debug(f"Prompt encode: time={(time.time() - t0):.3f}") return