Some work on caching text embeddings

2026-04-25 16:59:22 +00:00 · 2025-07-26 09:22:04 -06:00
parent 0d89c44624
commit 77dc38a574
4 changed files with 7 additions and 2 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -3020,6 +3020,8 @@ class StableDiffusion:
            active_modules = ['vae']
        if device_state_preset in ['cache_clip']:
            active_modules = ['clip']
+        if device_state_preset in ['cache_text_encoder']:
+            active_modules = ['text_encoder']
        if device_state_preset in ['unload']:
            active_modules = []
        if device_state_preset in ['generate']: