share text encoder

app.py

@@ -173,8 +173,9 @@ def load_pipeline(use_distilled: bool = False, device: str = "cuda"):
             enable_refiner_offloading=True
         )
         pipeline.to('cpu')
-        # refiner_pipeline = pipeline.refiner_pipeline
-        # refiner_pipeline.to('cpu')
+        refiner_pipeline = pipeline.refiner_pipeline
+        refiner_pipeline.text_encoder = pipeline.text_encoder
+        refiner_pipeline.to('cpu')
         reprompt_model = pipeline.reprompt_model
 
         print("✓ Pipeline loaded successfully")

hyimage/diffusion/pipelines/hunyuanimage_pipeline.py

@@ -816,8 +816,8 @@ class HunyuanImagePipeline:
         self.device = device
         if self.dit is not None:
             self.dit = self.dit.to(device, non_blocking=True)
-        if self.text_encoder is not None:
-            self.text_encoder = self.text_encoder.to(device, non_blocking=True)
+        # if self.text_encoder is not None:
+            # self.text_encoder = self.text_encoder.to(device, non_blocking=True)
         if self.vae is not None:
             self.vae = self.vae.to(device, non_blocking=True)
         return self