diff --git a/modules/model_lumina.py b/modules/model_lumina.py
index c3f5fb8d9..f4f79d833 100644
--- a/modules/model_lumina.py
+++ b/modules/model_lumina.py
@@ -2,6 +2,7 @@ import os
 import transformers
 import diffusers
 from huggingface_hub import repo_exists
+from modules import sd_hijack_te
 
 
 def load_lumina(_checkpoint_info, diffusers_load_config={}):
@@ -55,5 +56,6 @@ def load_lumina2(checkpoint_info, diffusers_load_config={}):
         **load_config,
     )
 
+    sd_hijack_te.init_hijack(pipe)
     devices.torch_gc(force=True)
     return pipe