From 4fa48e408495d82eb3005de216bfa43385ea7f57 Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Sun, 15 Jun 2025 22:19:56 +0300
Subject: [PATCH] Use te hijcak for with lumina 2

---
 modules/model_lumina.py | 2 ++
 1 file changed, 2 insertions(+)

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