Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

clear embeds cache on te change

Browse Source
This commit is contained in:
Vladimir Mandic
2024-09-18 21:21:30 -04:00
parent 3eb9e1dbb7
commit 51e97b3e43
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/model_te.py
+4
View File
@@ -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)