From 51e97b3e43b33c1e2548e5d88b3abf99c5dfe3ca Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 18 Sep 2024 21:21:30 -0400 Subject: [PATCH] clear embeds cache on te change --- modules/model_te.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules/model_te.py b/modules/model_te.py index 013806d18..367db1af4 100644 --- a/modules/model_te.py +++ b/modules/model_te.py @@ -124,6 +124,8 @@ def set_te(pipe): loaded_te = shared.opts.sd_text_encoder pipe.text_encoder = te.to(dtype=devices.dtype) shared.log.info(f'Load module: type="text_encoder" class="ViT-L" file="{shared.opts.sd_text_encoder}"') + import modules.prompt_parser_diffusers + modules.prompt_parser_diffusers.cache.clear() move_model(pipe.text_encoder, devices.device) if 'vit-g' in shared.opts.sd_text_encoder.lower() and hasattr(shared.sd_model, 'text_encoder_2') and shared.sd_model.text_encoder_2.__class__.__name__ == 'CLIPTextModelWithProjection': try: @@ -140,6 +142,8 @@ def set_te(pipe): loaded_te = shared.opts.sd_text_encoder pipe.text_encoder_2 = te.to(dtype=devices.dtype) shared.log.info(f'Load module: type="text_encoder_2" class="ViT-G" file="{shared.opts.sd_text_encoder}"') + import modules.prompt_parser_diffusers + modules.prompt_parser_diffusers.cache.clear() move_model(pipe.text_encoder_2, devices.device)