From 34c1bdad6e4fbcc5a52e9f4ce8ee21e569eaec5b Mon Sep 17 00:00:00 2001 From: GiusTex <112352961+GiusTex@users.noreply.github.com> Date: Sat, 5 Oct 2024 15:05:55 +0200 Subject: [PATCH] quick fix when updating the prompt text_encoders and tokenizers are not deleted when encoding prompt, and deleting them that way didn't free much vram anyway --- nodes.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/nodes.py b/nodes.py index 9557277..7bc58a3 100644 --- a/nodes.py +++ b/nodes.py @@ -274,13 +274,6 @@ class EncodeDiffusersOutpaintPrompt: "pooled_prompt_embeds": pooled_prompt_embeds, "negative_pooled_prompt_embeds": negative_pooled_prompt_embeds } - - del pipe.text_encoder, pipe.text_encoder_2, pipe.tokenizer, pipe.tokenizer_2 - gc.collect() - torch.cuda.empty_cache() - torch.cuda.ipc_collect() - # Update pipe - diffusers_outpaint_pipe["pipe"] = pipe return (diffusers_outpaint_pipe,diffusers_outpaint_conditioning,)