diff --git a/toolkit/models/base_model.py b/toolkit/models/base_model.py
index 9aadeea..243fd8a 100644
--- a/toolkit/models/base_model.py
+++ b/toolkit/models/base_model.py
@@ -1567,6 +1567,8 @@ class BaseModel:
             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 ['generate']:
             active_modules = ['vae', 'unet',
                               'text_encoder', 'adapter', 'refiner_unet']