Add gc to prompt parser

2026-09-19 09:14:35 +02:00 · 2024-06-23 04:50:30 +03:00
parent 6ad519e51f
commit 4d0ba8692a
1 changed files with 1 additions and 0 deletions
@@ -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